昇腾社区首页
中文
注册

SetShape

功能说明

设置matmul计算时的单次计算的形状singleM、singleN、singleK,单位为元素个数。

函数原型

int32_t SetShape(int32_t singleM, int32_t singleN, int32_t singleK)

参数说明

表1 参数说明

参数名

输入/输出

描述

singleM

输入

设置单次计算的singleM大小,单位为元素。

singleN

输入

设置单次计算的singleN大小,单位为元素。

singleK

输入

设置单次计算的singleK小,单位为元素。

返回值

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

注意事项