CANN/asc-devkit Conv3D模板参数

Conv3D模板参数【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit产品支持情况Ascend 950PR/Ascend 950DT不支持Atlas A3 训练系列产品/Atlas A3 推理系列产品支持Atlas A2 训练系列产品/Atlas A2 推理系列产品支持Atlas 200I/500 A2 推理产品不支持Atlas 推理系列产品AI Core不支持Atlas 推理系列产品Vector Core不支持Atlas 训练系列产品不支持功能说明创建Conv3D对象时需要传入Input、Weight、Output和Bias可选的参数类型信息类型信息通过ConvType来定义包括内存逻辑位置、数据格式、数据类型。Conv3dParam信息可选用于开启不同场景的性能优化模板。当前暂不支持使用。函数原型template class INPUT_TYPE, class WEIGHT_TYPE, class OUTPUT_TYPE, class BIAS_TYPE biasType, class CONV_CFG Conv3dParam using Conv3D Conv3dIntfExtConfigConvApi::ConvDataTypeINPUT_TYPE, WEIGHT_TYPE, OUTPUT_TYPE, BIAS_TYPE, CONV_CFG, Impl, Intf参数说明表1模板参数说明参数名输入/输出描述INPUT_TYPE输入ConvType类型模板参数指定Input的参数类型信息。WEIGHT_TYPE输入ConvType类型模板参数指定Weight的参数类型信息。OUTPUT_TYPE输入ConvType类型模板参数指定Output的参数类型信息。BIAS_TYPE可选输入ConvType类型模板参数指定Bias的参数类型信息。CONV_CFG可选输入ConvParam类型模板参数用于开启不同场景的性能优化模板当前版本只支持基础模板不开启性能优化。返回值说明无约束说明无调用示例#include lib/conv/conv3d/conv3d_api.h using inputType ConvApi::ConvTypeAscendC::TPosition::GM, ConvFormat::NDC1HWC0, bfloat16_t; using weightType ConvApi::ConvTypeAscendC::TPosition::GM, ConvFormat::FRACTAL_Z_3D, bfloat16_t; using outputType ConvApi::ConvTypeAscendC::TPosition::GM, ConvFormat::NDC1HWC0, bfloat16_t; // 可选参数如果不带Bias场景可以不传 using biasType ConvApi::ConvTypeAscendC::TPosition::GM, ConvFormat::ND, float; struct ConvCustom : public ConvApi::ConvParam { __aicore__ inline ConvCustom(){}; }; // 可选参数当前版本只支持基础模板不开启性能优化可以不传 Conv3dApi::Conv3DinputType, weightType, outputType, biasType, ConvCustom conv3dApi;【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考