昇腾社区首页
中文
注册

SetStride

功能说明

设置Stride信息。

函数原型

1
void SetStride(int64_t strideD, int64_t strideH, int64_t strideW)

参数说明

表1 参数说明

参数名

输入/输出

描述

strideD

输入

卷积正向过程中Depth方向Stride的大小。

strideH

输入

卷积正向过程中Height方向Stride的大小。

strideW

输入

卷积正向过程中Width方向Stride的大小。

返回值说明

约束说明

调用示例

1
2
3
4
optiling::Conv3DBackpropFilterTilingData tilingData;
auto ascendcPlatform = platform_ascendc::PlatformAscendCManager::GetInstance();
ConvBackpropApi::Conv3dBpFilterTiling conv3dBpDwTiling(*ascendcPlatform);
conv3dBpDwTiling.SetStride(strideD, strideH, strideW);