TensorSerializer is the serializer for Tensors. More...
#include <blob_serialization.h>
Public Member Functions | |
void | Serialize (const void *pointer, TypeMeta typeMeta, const string &name, SerializationAcceptor acceptor) override |
Serializes a Blob. More... | |
void | SerializeWithChunkSize (const void *pointer, TypeMeta typeMeta, const string &name, SerializationAcceptor acceptor, int chunk_size) override |
void | Serialize (const Tensor &tensor, const string &name, TensorProto *proto, size_t chunkBegin, int32_t chunkSize) |
Public Member Functions inherited from caffe2::BlobSerializerBase | |
virtual void | Serialize (const void *pointer, TypeMeta typeMeta, const std::string &name, SerializationAcceptor acceptor)=0 |
The virtual function that returns a serialized string for the input blob. More... | |
virtual void | SerializeWithChunkSize (const void *pointer, TypeMeta typeMeta, const std::string &name, SerializationAcceptor acceptor, int) |
Additional Inherited Members | |
Public Types inherited from caffe2::BlobSerializerBase | |
using | SerializationAcceptor = std::function< void(const std::string &blobName, const std::string &data)> |
TensorSerializer is the serializer for Tensors.
TensorSerializer takes in a blob that contains a Tensor, and serializes it into a TensorProto protocol buffer.
Definition at line 86 of file blob_serialization.h.
|
override |
Serializes a Blob.
Note that this blob has to contain Tensor, otherwise this function produces a fatal error.
Definition at line 104 of file blob_serialization.cc.