tvm
Classes | Static Public Member Functions | Public Attributes | List of all members
tvm::runtime::vm::TensorCacheMetadata Struct Reference

Metadata for Tensor cache, which by default, is named as "tensor-cache.json". More...

#include <tensor_cache_support.h>

Collaboration diagram for tvm::runtime::vm::TensorCacheMetadata:

Classes

struct  FileRecord
 Each shard of Tensor cache, which by default, is named as "params_shard_x.bin". More...
 

Static Public Member Functions

static TensorCacheMetadata Load (const std::string &path)
 Load the metadata from a specific directory. More...
 
static TensorCacheMetadata LoadFromStr (const std::string &json_str, const std::string &path)
 Load the metadata from a given JSON string. More...
 

Public Attributes

std::vector< FileRecordrecords
 The files in the Tensor cache. More...
 
std::string path
 The path to the tensor-cache.json file. More...
 

Detailed Description

Metadata for Tensor cache, which by default, is named as "tensor-cache.json".

Member Function Documentation

◆ Load()

static TensorCacheMetadata tvm::runtime::vm::TensorCacheMetadata::Load ( const std::string &  path)
static

Load the metadata from a specific directory.

◆ LoadFromStr()

static TensorCacheMetadata tvm::runtime::vm::TensorCacheMetadata::LoadFromStr ( const std::string &  json_str,
const std::string &  path 
)
static

Load the metadata from a given JSON string.

Member Data Documentation

◆ path

std::string tvm::runtime::vm::TensorCacheMetadata::path

The path to the tensor-cache.json file.

◆ records

std::vector<FileRecord> tvm::runtime::vm::TensorCacheMetadata::records

The files in the Tensor cache.


The documentation for this struct was generated from the following file: