TileOpFormat#
产品支持情况#
产品 |
是否支持 |
|---|---|
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)操作后得到的格式