Trait caffe2_opt::Converter

source · 
pub trait Converter {
    // Required method
    fn convert_to_neural_net_operator(
        &mut self,
        op: &OperatorDef
    ) -> Box<NeuralNetOperator>;

    // Provided methods
    fn get_arguments_from_operator(
        &self,
        op: OperatorDef
    ) -> HashMap<String, Argument> { ... }
    fn convert_to_operator_def(
        &mut self,
        nn_op: *const NeuralNetOperator
    ) -> OperatorDef { ... }
    fn get_device_option(&self, nn_op: *const NeuralNetOperator) -> DeviceOption { ... }
    fn get_kernel_shape(
        &mut self,
        arg_map: HashMap<String, Argument>
    ) -> Vec<i32> { ... }
}

Required Methods§

source

fn convert_to_neural_net_operator( &mut self, op: &OperatorDef ) -> Box<NeuralNetOperator>

Provided Methods§

source

fn get_arguments_from_operator( &self, op: OperatorDef ) -> HashMap<String, Argument>

source

fn convert_to_operator_def( &mut self, nn_op: *const NeuralNetOperator ) -> OperatorDef

source

fn get_device_option(&self, nn_op: *const NeuralNetOperator) -> DeviceOption

source

fn get_kernel_shape(&mut self, arg_map: HashMap<String, Argument>) -> Vec<i32>

Implementors§