TensorSerializer is the serializer for Tensors. More...

Inheritance diagram for caffe2::TensorSerializer< Context >:

Public Member Functions
void	Serialize (const Blob &blob, const string &name, SerializationAcceptor acceptor) override
	Serializes a Blob. More...

void	SerializeWithChunkSize (const Blob &blob, const string &name, SerializationAcceptor acceptor, int chunk_size) override

void	Serialize (const Tensor< Context > &tensor, const string &name, TensorProto *proto, size_t chunkBegin, int32_t chunkSize)

Public Member Functions inherited from caffe2::BlobSerializerBase
virtual void	Serialize (const Blob &blob, const std::string &name, SerializationAcceptor acceptor)=0
	The virtual function that returns a serialized string for the input blob. More...

virtual void	SerializeWithChunkSize (const Blob &blob, const std::string &name, SerializationAcceptor acceptor, int chunk_size)

Additional Inherited Members
Public Types inherited from caffe2::BlobSerializerBase
using	SerializationAcceptor = std::function< void(const std::string &blobName, const std::string &data)>

Detailed Description

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 42 of file blob_serialization.h.

Member Function Documentation

template<class Context >

void caffe2::TensorSerializer< Context >::Serialize	(	const Blob &	blob,
		const string &	name,
		BlobSerializerBase::SerializationAcceptor	acceptor
	)

override

Serializes a Blob.

Note that this blob has to contain Tensor<Context>, otherwise this function produces a fatal error.

Definition at line 182 of file blob_serialization.h.

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