Apache SINGA
A distributed deep learning platform .
 All Classes Namespaces Files Functions Variables Typedefs Enumerator Macros
Classes | Namespaces | Functions
tensor_io.h File Reference

definitions of I/O functions for mshadow tensor More...

#include <cstdio>
#include "tensor.h"

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...
 

Detailed Description

definitions of I/O functions for mshadow tensor

Author
Tianqi Chen