CacheDesc
Cache的描述信息
1 2 3 4 5 6 7 | struct CacheDesc {
CachePlacement placement = CachePlacement::kDevice; // 内存类型
uint32_t num_tensors = 0U; // Cache包含的tensor个数
DataType data_type = DT_UNDEFINED; // Cache中tensor的数据类型
std::vector<int64_t> shape; // Cache中tensor的shape
uint8_t reserved[128]; // 预留
}
|
父主题: 数据结构