mxnet
Classes | Namespaces | Functions
io.h File Reference
#include "./tensor.h"
Include dependency graph for io.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  mshadow::utils::IStream
 interface of stream I/O, used to serialize data, mshadow does not restricted to only this interface in SaveBinary/LoadBinary mshadow accept all class that implements Read and Write More...
 

Namespaces

 mshadow
 namespace for mshadow
 
 mshadow::utils
 

Functions

template<int dim, typename DType , typename TStream >
void mshadow::SaveBinary (TStream &fo, const Tensor< cpu, dim, DType > &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 DType , typename TStream >
void mshadow::SaveBinary (TStream &fo, const Tensor< gpu, dim, DType > &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 DType , typename TStream >
void mshadow::LoadBinary (TStream &fi, Tensor< cpu, dim, DType > *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 DType , typename TStream >
void mshadow::LoadBinary (TStream &fi, Tensor< gpu, dim, DType > *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...