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