昇腾社区首页
中文
注册

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);