设置A矩阵的位置，数据格式，数据类型，是否转置等信息。
int32_t SetAType(TPosition pos, CubeFormat type, DataType dataType, bool isTrans = false);
参数名 |
输入/输出 |
描述 |
|---|---|---|
pos |
输入 |
A矩阵所在的buffer位置，可设置为：TPosition::GM/TPosition::TSCM。 |
type |
输入 |
A矩阵的数据格式。CubeFormat枚举定义如下： enum class CubeFormat {
ND = 0,
NZ,
ZN,
ZZ,
NN,
ND_ALIGN
};
|
dataType |
输入 |
A矩阵的数据类型，可设置为：DataType::DT_FLOAT/DataType::DT_FLOAT16。 |
isTrans |
输入 |
A矩阵是否转置。 |
-1表示设置失败； 0表示设置成功。
无