SetDilation
功能说明
设置Dilation信息。
函数原型
1 | void SetDilation(int64_t dilationD, int64_t dilationH, int64_t dilationW) |
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
dilationD |
输入 |
D方向Dilation大小。 |
dilationH |
输入 |
H方向Dilation大小。 |
dilationW |
输入 |
W方向Dilation大小。 |
返回值说明
无
约束说明
在调用GetTiling接口前,本接口可选调用。若未调用本接口,默认dilationD=1, dilationH=1, dilationW=1。
调用示例
1 2 3 4 | // 实例化Conv3D Api auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo()); Conv3dTilingApi::Conv3dTiling conv3dApiTiling(ascendcPlatform); conv3dApiTiling.SetDilation(dilationD, dilationH, dilationW); |
父主题: Conv3D Tiling