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