在不改变tiling的情况下,重新设置本次计算的singleCoreM/singleCoreN/singleCoreK。
__aicore__ inline void SetTail(int tailM = -1, int tailN = -1, int tailK = -1)
参数名 |
输入/输出 |
描述 |
---|---|---|
tailM |
输入 |
重新设置的singleCoreM值 |
tailN |
输入 |
重新设置的singleCoreN值 |
tailK |
输入 |
重新设置的singleCoreK值 |
无
Atlas A2训练系列产品/Atlas 800I A2推理产品
Atlas推理系列产品AI Core
Atlas 200I/500 A2推理产品
无