Struct caffe2_nomnigraph::NeuralNetOperator

pub struct NeuralNetOperator { /* private fields */ }

Implementations

impl NeuralNetOperator

pub fn new_from_kind_opcode_layout(k: NNKind, i: Opcode, l: NNLayout) -> Self

pub fn new_from_kind_and_opcode(k: NNKind, i: Opcode) -> Self

pub fn new_from_kind_and_layout(k: NNKind, l: NNLayout) -> Self

pub fn get_kind(&self) -> NNKind

pub fn set_layout(&mut self, l: NNLayout)

pub fn get_layout(&self) -> NNLayout

pub fn set_annotation(&mut self, extra_annotation: Box<Annotation>)

pub fn get_annotation(&self) -> *const Annotation

pub fn get_mutable_annotation(&mut self) -> *mut Annotation

pub fn check_inputs_and_outputs( &mut self, inputs: Vec<*const NeuralNetData>, outputs: Vec<*const NeuralNetData> ) -> bool

| ———– | @brief | | Validate the inputs and outputs to this | operator. | | \p inputs A vector of references to | | NeuralNetData types that represent | the data being fed into the operator. | | \p outputs A vector of references to | | NeuralNetData types that represent | the data being outputted by the operator. | | | ———– | @return | | true if the inputs and outputs are compatible | with the operator. |

Trait Implementations

impl Default for NeuralNetOperator

fn default() -> Self

Returns the “default value” for a type.

impl From<NNKind> for NeuralNetOperator

fn from(k: NNKind) -> Self

Converts to this type from the input type.

impl Named for NeuralNetOperator

fn get_name(&self) -> String

impl<N> get_impl<N> for NeuralNetOperator

fn impl_(n: N) -> *mut Self

impl<N> is_impl<N> for NeuralNetOperator

fn impl_(n: N) -> bool