GetTensorType
函数功能
用于获取Tensor的内存类型。
函数原型
MemoryData::MemoryType GetTensorType() const;
返回参数说明
数据结构 |
说明 |
---|---|
MemoryType |
申请的内存类型: ● MEMORY_HOST:对应Host侧。 ● MEMORY_DEVICE:对应Device侧。 ● MEMORY_DVPP:对应DVPP侧。 ● MEMORY_HOST_MALLOC:对应malloc申请内存。 ● MEMORY_HOST_NEW:对应new申请内存。 |
父主题: TensorBase