Struct caffe2_proto::ArgumentHelper

pub struct ArgumentHelper { /* private fields */ }

| ———– | @brief | | A helper class to index into arguments. | | This helper helps us to more easily index | into a set of arguments that are present | in the operator. | | To save memory, the argument helper | does not copy the operator def, so one | would need to make sure that the lifetime | of the OperatorDef object outlives | that of the ArgumentHelper. |

Implementations

impl ArgumentHelper

pub fn new(def: &OperatorDef) -> Self

pub fn new_from_netdef(netdef: &NetDef) -> Self

pub fn has_argument(&self, name: &String) -> bool

pub fn has_argument_with_def<Def>(def: &Def, name: &String) -> bool

pub fn get_single_argument<Def, T>(def: &Def, name: &str, default_value: T) -> T

pub fn has_single_argument_of_type<Def, T>(def: &Def, name: &String) -> bool

pub fn get_repeated_argument<Def, T>( def: &Def, name: &String, default_value: &Vec<T> ) -> Vec<T>

pub fn get_message_argument_with_def<Def, MessageType>( def: &Def, name: &String ) -> MessageType

pub fn get_repeated_message_argument_with_def<Def, MessageType>( def: &Def, name: &String ) -> Vec<MessageType>

pub fn remove_argument<Def>(def: &mut Def, index: i32) -> bool

pub fn get_message_argument<MessageType>(&self, name: &String) -> MessageType

pub fn get_repeated_message_argument<MessageType>( &self, name: &String ) -> Vec<MessageType>