昇腾社区首页
中文
注册

GetTensorType

函数功能

用于获取Tensor的内存类型。

函数原型

MemoryData::MemoryType GetTensorType() const;

返回参数说明

数据结构

说明

MemoryType

申请的内存类型:

● MEMORY_HOST:对应Host侧。

● MEMORY_DEVICE:对应Device侧。

● MEMORY_DVPP:对应DVPP侧。

● MEMORY_HOST_MALLOC:对应malloc申请内存。

● MEMORY_HOST_NEW:对应new申请内存。