设置Dilation信息,即卷积核Depth/Height/Width方向的扩张大小。
1 | void SetDilation(int64_t dilationD, int64_t dilationH, int64_t dilationW) |
参数名 |
输入/输出 |
描述 |
|---|---|---|
dilationD |
输入 |
卷积核Weight Depth方向扩张大小。 |
dilationH |
输入 |
卷积核Weight Height方向扩张大小。 |
dilationW |
输入 |
卷积核Weight Width方向扩张大小。 |
无
无
1 2 3 4 | optiling::Conv3DBackpropFilterTilingData tilingData; auto ascendcPlatform = platform_ascendc::PlatformAscendCManager::GetInstance(); ConvBackpropApi::Conv3dBpFilterTiling conv3dBpDwTiling(*ascendcPlatform); conv3dBpDwTiling.SetDilation(dilationD, dilationH, dilationW); |