设置Stride信息。
1 | void SetStride(int64_t strideD, int64_t strideH, int64_t strideW) |
参数名 |
输入/输出 |
描述 |
|---|---|---|
strideD |
输入 |
D方向Stride大小。 |
strideH |
输入 |
H方向Stride大小。 |
strideW |
输入 |
W方向Stride大小。 |
无
在调用GetTiling接口前,本接口可选调用。若未调用本接口,默认strideD=1, strideH=1, strideW=1。
1 2 3 4 | // 实例化Conv3D Api auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo()); Conv3dTilingApi::Conv3dTiling conv3dApiTiling(ascendcPlatform); conv3dApiTiling.SetStride(strideD, strideH, strideW); |