Struct caffe2_opt::Caffe2Annotation

pub struct Caffe2Annotation<T, U> { /* private fields */ }

Implementations

impl<T, U> Caffe2Annotation<T, U>

pub fn new(device: String) -> Self

pub fn set_operator_def(&mut self, op_def: &OperatorDef)

pub fn has_operator_def(&self) -> bool

pub fn get_operator_def(&self) -> &OperatorDef

pub fn get_mutable_operator_def(&mut self) -> *mut OperatorDef

pub fn set_device_option(&mut self, dev_opt: &DeviceOption)

Distributed annotations

pub fn has_device_option(&self) -> bool

pub fn get_device_option(&self) -> &DeviceOption

pub fn get_mutable_device_option(&mut self) -> *mut DeviceOption

pub fn set_device(&mut self, device: String)

pub fn get_device(&self) -> String

pub fn set_device_type(&mut self, device: i32)

pub fn get_device_type(&self) -> i32

pub fn set_parallelization(&mut self, s: ParallelizationScheme, num: i32)

pub fn get_parallelization_scheme(&self) -> ParallelizationScheme

pub fn get_parallelization(&self) -> i32

pub fn set_key_node(&mut self, n: NodeRef<T, U>)

pub fn get_key_node(&self) -> &NodeRef<T, U>

pub fn set_length_node(&mut self, n: NodeRef<T, U>)

pub fn get_length_node(&self) -> &NodeRef<T, U>

pub fn set_component_levels(&mut self, components: Vec<String>)

pub fn get_component_levels(&self) -> Vec<String>

pub fn classof(&mut self, a: *const Annotation) -> bool

Trait Implementations

impl<T, U> Default for Caffe2Annotation<T, U>

fn default() -> Self

Returns the “default value” for a type.