CANN/asc-devkit Conv3DBackpropInput GetTiling函数

GetTiling【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit功能说明获取Tiling参数。函数原型int64_t GetTiling(optiling::Conv3DBackpropInputTilingData tiling)int64_t GetTiling(AscendC::tiling::Conv3DBackpropInputTilingData tiling)参数说明表1参数说明参数名输入/输出描述tiling输出Conv3DBackpropInput的Tiling结构体用于存储最终的Tiling结果。TConv3DBackpropInputTiling结构介绍请参考TConv3DApiTiling结构体说明。返回值说明如果返回值不为-1则代表Tiling计算成功用户可以使用该Tiling结构的值。如果返回值为-1则代表Tiling计算失败该Tiling结果无法使用。约束说明无调用示例// 构建Conv3dBackpropInput算子tiling的调用示例 auto ascendcPlatform platform_ascendc::PlatformAscendC(context-GetPlatformInfo()); ConvBackpropApi::Conv3DBpInputTiling tiling(ascendcPlatform); tiling.SetWeightShape(256, 128, 3, 3, 3); tiling.SetInputShape(1, 128, 11, 128, 128); tiling.SetGradOutputShape(1, 256, 9, 128, 128); tiling.SetPadding(0, 0, 1, 1, 1, 1); tiling.SetDilation(1, 1, 1); tiling.SetStride(1, 1, 1); optiling::Conv3DBackpropInputTilingData tilingData; int ret tiling.GetTiling(tilingData); // 获取Tiling参数 AscendC::tiling::Conv3DBackpropInputTilingData tilingDataNotOp; ret tiling.GetTiling(tilingDataNotOp); // 使用AscendC::tiling::Conv3DBackpropInputTilingData获取Tiling参数// 构建Conv3dTranspose算子tiling的调用示例 auto ascendcPlatform platform_ascendc::PlatformAscendC(context-GetPlatformInfo()); ConvBackpropApi::Conv3DBpInputTiling tiling(ascendcPlatform); tiling.SetWeightShape(256, 128, 3, 3, 3); tiling.SetGradOutputShape(256, 9, 128, 128); // 等价于Conv3dTranspose的输入X tiling.SetPadding(0, 0, 1, 1, 1, 1); tiling.SetDilation(1, 1, 1); tiling.SetStride(1, 1, 1); tiling.SetOutputPadding(0, 0, 0); // 对Conv3dTranspose的输出Y进行padding optiling::Conv3DBackpropInputTilingData tilingData; int ret tiling.GetTiling(tilingData); // 获取Tiling参数【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考