昇腾社区首页
中文
注册

SetSingleShape

功能说明

设置Matmul单核计算的形状singleMIn,singleNIn,singleKIn,单位为元素个数。

函数原型

virtual int32_t SetSingleShape(int32_t singleMIn = -1, int32_t singleNIn = -1, int32_t singleKIn = -1);

参数说明

表1 参数说明

参数名

输入/输出

描述

singleMIn

输入

设置的singleMIn大小,单位为元素,默认值为-1。

singleNIn

输入

设置的singleNIn大小,单位为元素,默认值为-1。

singleKIn

输入

设置的singleKIn大小,单位为元素,默认值为-1。

返回值

-1表示设置失败; 0表示设置成功。

注意事项