Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
Serialization.Tensor Module Reference

Classes

struct  Mila::Dnn::Serialization::ITensorBlob
 Type-erased interface for a serialized tensor blob. More...
struct  Mila::Dnn::Serialization::TensorBlob< MR >
 Concrete tensor blob owning a TensorBuffer-backed raw byte buffer. More...
 Metadata describing a tensor in serialized form. More...

Functions

template<typename MR = Compute::CpuMemoryResource>
requires isValidTensor<dtype_t::UINT8, MR>
TensorBlob< MR > Mila::Dnn::Serialization::readTensorBlob (const ModelArchive &archive, const std::string &prefix, int device_id=0)
 Read tensor metadata and raw bytes from prefix into a typed blob.
void Mila::Dnn::Serialization::writeTensorBlob (ModelArchive &archive, const std::string &prefix, const TensorMetadata &meta, const void *data, size_t size)
 Write tensor metadata and raw bytes under the given prefix into archive.

Files

file  /__w/Mila/Mila/Mila/Src/Dnn/Tensors/Tensor.Serialization.ixx
 Tensor-specific serialization helpers and metadata.