昇腾社区首页
中文
注册

简介

TilingData用于存储算子Tiling的计算结果,即Tiling实现根据算子的shape等信息确定的数据切分算法相关参数(比如每次搬运的数据块大小,搬运、计算操作的循环次数)

需要包含的头文件

1
#include <tensor_data.h>

Public成员函数

TilingData(const TilingData &) = delete
TilingData(TilingData &&) = delete
TilingData operator=(const TilingData &) = delete
TilingData operator=(TilingData &&) = delete
size_t GetCapacity() const
size_t GetDataSize() const
void SetDataSize(const size_t size)
void *GetData()
const void *GetData() const
template<typename T, typename std::enable_if<std::is_standard_layout<T>::value, int>::type = 0>
ge::graphStatus Append(const T &data)
template<typename T, typename std::enable_if<std::is_standard_layout<T>::value, int>::type = 0>
ge::graphStatus Append(const T *data, size_t append_num)
static std::unique_ptr<uint8_t[]> CreateCap(const size_t cap_size)
static ge::graphStatus CalcTotalSize(const size_t cap_size, size_t &total_size)
void Init(const size_t cap_size, void *const data)
ge::graphStatus AppendConvertedAttrVal(const RuntimeAttrs *attrs, const size_t attr_index, const AttrDataType src_type, const AttrDataType dst_type)
template<typename T>
TilingData &operator<<(TilingData &out, const T &data)