Struct caffe2op_loadsave::LoadOp

pub struct LoadOp<Context> { /* private fields */ }

| The Load operator loads a set of serialized | blobs from a db or multiple dbs. It takes | $[0, \infty)$ number of inputs and $[0, | \infty)$ number of outputs, using the | db keys to match the db entries with the | outputs. | | If at least one input is passed, then | it is assumed that that input blobs are | a set of DBReaders to load from. Otherwise | the db or dbs argument is used to | load blobs from one single db or multiple | dbs respectively. db_type argument | is used to specify the type of the input | db/dbs. | | Github Links: | | - https://github.com/pytorch/pytorch/blob/master/caffe2/operators/load_save_op.cc |

Implementations

impl<Context> LoadOp<Context>

pub fn new(operator_def: &OperatorDef, ws: *mut Workspace) -> Self

pub fn run_on_device(&mut self) -> bool

pub fn extract( &mut self, db_id: i32, cursor: *mut dyn Cursor, blob_states: *mut HashMap<String, BlobState>, total_loaded_blobs: *mut i32 )

pub fn extract_all( &mut self, db_id: i32, cursor: *mut dyn Cursor, blob_states: *mut HashMap<String, BlobState>, total_loaded_blobs: *mut i32 )

pub fn extract_from( &mut self, db_id: i32, cursor: *mut dyn Cursor, outputs: &Vec<*mut Blob>, blob_states: *mut HashMap<String, BlobState>, total_loaded_blobs: *mut i32 )

impl LoadOp<CPUContext>

pub fn set_current_device(&mut self, proto: *mut BlobProto)

impl LoadOp<CUDAContext>

pub fn set_current_device(&mut self, proto: *mut BlobProto)