|
Mila 0.13.48
Deep Neural Network Library
|
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... | |
| struct | Mila::Dnn::Serialization::TensorMetadata |
| 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. | |