简介
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)
父主题: TensorData