Apache SINGA
A distributed deep learning platform .
|
definitions of I/O functions for mshadow tensor More...
Go to the source code of this file.
Classes | |
class | mshadow::utils::IStream |
interface of stream I/O, used to serialize data, it is not restricted to only this interface in SaveBinary/LoadBinary mshadow accept all class that implements Read and Write More... | |
class | mshadow::utils::FileStream |
implementation of file i/o stream More... | |
Namespaces | |
mshadow | |
namespace for mshadow | |
mshadow::utils | |
namespace for helper utils of the project | |
Functions | |
template<int dim, typename TStream > | |
void | mshadow::SaveBinary (TStream &fo, const Tensor< cpu, dim > &src) |
CPU/GPU: save a tensor by binary format, for GPU version, a temp Tensor<cpu,dim> storage will be allocated. More... | |
template<int dim, typename TStream > | |
void | mshadow::SaveBinary (TStream &fo, const Tensor< gpu, dim > &src) |
refer to comment of cpu ver More... | |
template<int dim, typename TStream > | |
void | mshadow::LoadBinary (TStream &fi, Tensor< cpu, dim > &dst, bool pre_alloc) |
CPU/GPU: load a tensor by binary format, for GPU version, a temp Tensor<cpu,dim> storage will be allocated if pre_alloc is true , then space in dst is preallocated, and must have same shape of the tensor loaded if pre_alloc is false, then dst originally does not have space allocated, LoadBinary will allocate space for dst. More... | |
template<int dim, typename TStream > | |
void | mshadow::LoadBinary (TStream &fi, Tensor< gpu, dim > &dst, bool pre_alloc) |
refer to comment of cpu ver More... | |
definitions of I/O functions for mshadow tensor