Conv3D模板参数

产品支持情况

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件	√
Atlas 200I/500 A2 推理产品	x
Atlas 推理系列产品AI Core	x
Atlas 推理系列产品Vector Core	x
Atlas 训练系列产品	x
Atlas 200/300/500 推理产品	x

功能说明

创建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 = Conv3dIntfExt<Config<ConvApi::ConvDataType<INPUT_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::ConvType<AscendC::TPosition::GM, ConvFormat::NDC1HWC0, bfloat16_t>;
using weightType = ConvApi::ConvType<AscendC::TPosition::GM, ConvFormat::FRACTAL_Z_3D, bfloat16_t>;
using outputType = ConvApi::ConvType<AscendC::TPosition::GM, ConvFormat::NDC1HWC0, bfloat16_t>;
using biasType = ConvApi::ConvType<AscendC::TPosition::GM, ConvFormat::ND, float>; // 可选参数，如果不带Bias场景，可以不传
struct ConvCustom : public ConvApi::ConvParam {
    __aicore__ inline ConvCustom(){};
}; // 可选参数，当前版本只支持基础模板，不使能性能优化，可以不传

Conv3dApi::Conv3D<inputType, weightType, outputType, biasType, ConvCustom> conv3dApi;

父主题： Conv3D