GetTensorType

Function Usage

Obtains the memory type of a tensor.

Prototype

MemoryData::MemoryType GetTensorType() const;

Return Parameter Description

Data Structure

Description

MemoryType

Type of the allocated memory:

MEMORY_HOST corresponds to the host.

MEMORY_DEVICE corresponds to the device.

MEMORY_DVPP corresponds to the DVPP.

MEMORY_HOST_MALLOC corresponds to the memory allocated by malloc.

MEMORY_HOST_NEW corresponds to the memory allocated by new.