SetStride
功能说明
设置Stride信息。
函数原型
Text
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。
调用示例
Text
// 实例化Conv3D Api
auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo());
Conv3dTilingApi::Conv3dTiling conv3dApiTiling(ascendcPlatform);
conv3dApiTiling.SetStride(strideD, strideH, strideW);