Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Friends
caffe2::serialize::PyTorchStreamReader Class Referencefinal

Public Member Functions

 PyTorchStreamReader (const std::string &file_name)
 
 PyTorchStreamReader (std::istream *in)
 
 PyTorchStreamReader (std::unique_ptr< ReadAdapterInterface > in)
 
std::tuple< at::DataPtr, size_t > getRecord (const std::string &name)
 
size_t getRecordOffset (const std::string &name)
 

Friends

size_t istream_read_func (void *pOpaque, uint64_t file_ofs, void *pBuf, size_t n)
 

Detailed Description

Definition at line 101 of file inline_container.h.


The documentation for this class was generated from the following files: