Conv3DBackpropInput Tiling使用说明【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkitAscend C提供一组Conv3DBackpropInput Tiling API方便用户获取Conv3DBackpropInput Kernel计算时所需的Tiling参数。用户只需要传入Input/GradOutput/Weight的Position位置、Format格式和DType数据类型及相关参数等信息调用API接口即可获取Init中TConv3DBackpropInputTiling结构体中的相关参数。Conv3DBackpropInput Tiling API提供一个GetTiling接口获取Tiling参数获取Tiling参数的流程如下创建一个单核Tiling对象。设置Input、GradOutput、Weight的参数类型信息以及Shape信息如果存在Padding、Stride参数通过SetPadding、SetStride接口设置。调用GetTiling接口获取Tiling信息。使用Conv3DBackpropInput Tiling接口获取Tiling参数的样例如下#include tiling/conv_backprop/conv3d_bp_input_tiling.h optiling::Conv3DBackpropInputTilingData tilingData; auto ascendcPlatform platform_ascendc::PlatformAscendCManager::GetInstance(); ConvBackpropApi::Conv3DBpInputTiling conv3DBpDxTiling(*ascendcPlatform); conv3DBpDxTiling.SetWeightType( Convolution3DBackprop::TPosition::GM, Convolution3DBackprop::ConvFormat::FRACTAL_Z_3D, Convolution3DBackprop::ConvDtype::FLOAT32); conv3DBpDxTiling.SetGradOutputType( Convolution3DBackprop::TPosition::GM, Convolution3DBackprop::ConvFormat::NDC1HWC0, Convolution3DBackprop::ConvDtype::FLOAT16); conv3DBpDxTiling.SetInputType( Convolution3DBackprop::TPosition::CO1, Convolution3DBackprop::ConvFormat::NDC1HWC0, Convolution3DBackprop::ConvDtype::FLOAT16); conv3DBpDxTiling.SetInputShape(orgN, orgCi, orgDi, orgHi, orgWi); conv3DBpDxTiling.SetGradOutputShape(orgCo, orgDo, orgHo, orgWo); conv3DBpDxTiling.SetWeightShape(orgKd, orgKh, orgKw); conv3DBpDxTiling.SetPadding(padFront, padBack, padUp, padDown, padLeft, padRight); conv3DBpDxTiling.SetStride(strideD, strideH, strideW); conv3DBpDxTiling.SetDilation(dilationD, dilationH, dilationW); int ret conv3DBpDxTiling.GetTiling(tilingData); // if ret -1, get tiling failed需要包含的头文件#include lib/conv_backprop/conv3d_bp_input_tiling.h【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考