SetSingleOutputShape
功能说明
设置单核上结果矩阵Output的形状。
函数原型
1 | void SetSingleOutputShape(int64_t singleCo, int64_t singleDo, int64_t singleM) |
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
singleCo |
输入 |
单核上输出通道的大小。 |
singleDo |
输入 |
单核上Output D维度大小。 |
singleM |
输入 |
单核上Output M维度大小。 |
返回值说明
无
约束说明
无
调用示例
1 2 3 4 | // 实例化Conv3D Api auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo()); Conv3dTilingApi::Conv3dTiling conv3dApiTiling(ascendcPlatform); conv3dApiTiling.SetSingleOutputShape(singleCo, singleDo, singleM); |
父主题： Conv3D Tiling