SetDilation
功能说明
设置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); |