Constructor and Destructor

Description

Constructs a TensorData.

Method 1: The address of the tensor data and the manager used to manage the tensor data are specified.

Method 2: The address of the tensor data, the manager used to manage the tensor data, the memory occupied by the tensor data, and the location (host or device) of the tensor data are specified.

Method 3: move constructor.

Note: If the manager is nullptr, addr is the address of the tensor data. Otherwise, the address of the tensor data is provided by the manager.

Prototype

• Constructor

  TensorData(TensorAddress addr = nullptr, const TensorAddrManager manager = nullptr)

  TensorData(TensorAddress addr, const TensorAddrManager manager, size_t size, TensorPlacement placement)

  TensorData(TensorData &&other) noexcept

  TensorData(const TensorData &) = delete

• Destructor

  ~TensorData

Parameters

using TensorAddress = void *;

using TensorAddrManager = ge::graphStatus (*)(TensorAddress addr, TensorOperateType operate_type, void **out);
enum TensorOperateType {
  kGetTensorAddress,  ///< Obtain the address of the tensor.
  kFreeTensor,        ///< Free the tensor.
  kPlusShareCount,    ///< Share the tensor.
  kTensorOperateType
};

Returns

Initialized TensorData object

Restrictions

None

Example

auto addr = reinterpret_cast<void *>(0x10);
TensorData td(addr, HostAddrManager, 100U, kOnHost);