TileOpFormat#
产品支持情况#
Ascend 950PR/Ascend 950DT:支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品:支持
Atlas A2 训练系列产品/Atlas A2 推理系列产品:支持
功能说明#
TileOpFormat定义了Tensor的瓦片操作格式,用于优化不同计算模式下的内存访问和计算效率。主要区分密集计算和稀疏计算的瓦片格式。
原型定义#
class TileOpFormat(enum.Enum):
TILEOP_ND = ... # N维Tensor,支持标准的多维数组操作
TILEOP_NZ = ... # 同FRACTAL_NZ/NZ是对一个Tensor最低两维(一个Tensor的所有维度,右侧为低维,左侧为高维)进行填充(pad)、拆分(reshape)和转置(transpose)操作后得到的格式