GetTensorType

Function Usage

Obtains the memory type of a tensor.

Prototype

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

Response Parameters

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.