SetDilation
功能说明
设置Dilation信息,即卷积核Depth/Height/Width方向的扩张大小。
函数原型
void SetDilation(int64_t dilationD, int64_t dilationH, int64_t dilationW)
参数说明
表 1 参数说明
返回值说明
无
约束说明
无
调用示例
optiling::Conv3DBackpropFilterTilingData tilingData;
auto ascendcPlatform = platform_ascendc::PlatformAscendCManager::GetInstance();
ConvBackpropApi::Conv3dBpFilterTiling conv3dBpDwTiling(*ascendcPlatform);
conv3dBpDwTiling.SetDilation(dilationD, dilationH, dilationW);