SetWeightShape

功能说明

设置权重矩阵Weight的形状。

函数原型

void SetWeightShape(int64_t cout, int64_t cin, int64_t d, int64_t h, int64_t w)

参数说明

表 1参数说明

返回值说明

无

约束说明

无

调用示例

optiling::Conv3DBackpropFilterTilingData tilingData; auto ascendcPlatform = platform_ascendc::PlatformAscendCManager::GetInstance(); ConvBackpropApi::Conv3dBpFilterTiling conv3dBpDwTiling(*ascendcPlatform); conv3dBpDwTiling.SetWeightShape(cout, cin, d, h, w);