GetTensorType

函数功能

用于获取Tensor的内存类型。

函数原型

1
MemoryData::MemoryType TensorBase::GetTensorType() const;

返回参数说明

数据结构

说明

MemoryType

申请的内存类型:

● MEMORY_HOST:对应Host侧。

● MEMORY_DEVICE:对应Device侧。

● MEMORY_DVPP:对应DVPP侧。

● MEMORY_HOST_MALLOC:对应malloc申请内存。

● MEMORY_HOST_NEW:对应new申请内存。