昇腾社区首页
中文
注册

简介

TensorData是一个张量数据管理类,用于记录张量的地址、大小、存放位置(TensorPlacement),支持张量的共享、所有权释放,管理张量的生命周期。

需要包含的头文件

1
#include <tensor_data.h>

Public成员函数

explicit TensorData(TensorAddress addr = nullptr, const TensorAddrManager manager = nullptr)
explicit TensorData(TensorAddress addr, const TensorAddrManager manager, size_t size, TensorPlacement placement)
TensorData(TensorData &&other) noexcept
TensorData(const TensorData &) = delete
~TensorData
TensorData& operator= (const TensorData &other)=delete
TensorData& operator= (TensorData &&other) noexcept
TensorAddress GetAddr() const
size_t GetSize() const
void SetSize(const size_t size)
TensorPlacement GetPlacement() const
void SetPlacement(const TensorPlacement placement)
ge::graphStatus Free()
ge::graphStatus SetAddr(const ConstTensorAddressPtr addr, TensorAddrManager manager)
bool IsSharedWith(const TensorData &other) const
ge::graphStatus ShareFrom(const TensorData &other)