aclnnConvTbc

支持的产品型号

[object Object]Atlas 推理系列产品[object Object]。
[object Object]Atlas 训练系列产品[object Object]。
[object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品[object Object]。

接口原型

每个算子分为，必须先调用“aclnnConvTbcGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器，再调用“aclnnConvTbc”接口执行计算。

[object Object]
[object Object]

功能描述

算子功能：实现输入输出维度为 T(时间或空间维度)、B(批次)、C(通道)的一维卷积。
计算公式：假定输入（input）的shape是( $H_{\text{in}},N,C_{\text{in}}$ )，输出（output）的shape是( $H_{\text{out}},N,C_{\text{out}}$ )，那输出将被表示为： $out_{N_i,C_{out j}} = bias(C_{out}) + \sum_{k = 0}^{C_{in} - 1} weight(:, k, C_{out j}) \cdot input(N_i, k)$ 其中 $N$ 表示batch_size（批次）， $C$ 表示通道数， $H$ 代表时间或空间维度。

aclnnConvTbcGetWorkspaceSize

参数说明：
- self(aclTensor*, 计算输入)：计算公式中的[object Object]，，类型与weight一致，支持，shape为( $H_{in},N,C_{in}$ $H_{in}, N, C_{in}$ )，为NCL。
  - [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品[object Object]：数据类型支持FLOAT，FLOAT16，BFLOAT16。
  - [object Object]Atlas 推理系列产品[object Object]、[object Object]Atlas 训练系列产品[object Object]：数据类型支持FLOAT，FLOAT16。
- weight(aclTensor*, 计算输入)：计算公式中的[object Object]，，类型与self一致，支持，shape为( $K,C_{\text{in}},C_{\text{out}}$ $K, C_{in}, C_{out}$ )，为NCL。
  - [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品[object Object]：数据类型支持FLOAT，FLOAT16，BFLOAT16。
  - [object Object]Atlas 推理系列产品[object Object]、[object Object]Atlas 训练系列产品[object Object]：数据类型支持FLOAT，FLOAT16。
- bias(aclTensor*, 计算输入)：计算公式中的[object Object]，，支持，shape为( $C_{\text{out}}$ $C_{out}$ )，为ND；bias传入的指针类型入参不允许是空指针。
  - [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品[object Object]：数据类型支持FLOAT，FLOAT16，BFLOAT16。
  - [object Object]Atlas 推理系列产品[object Object]、[object Object]Atlas 训练系列产品[object Object]：数据类型支持FLOAT，FLOAT16。
- pad(int64, 计算输入)：表示 H维度上左右的填充个数。
- output(aclTensor*, 计算输出)：计算公式中的[object Object]，，类型与self一致，支持，shape为( $H_{\text{out}},N,C_{\text{out}}$ $H_{out}, N, C_{out}$ )，为NCL。
  - [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品[object Object]：数据类型支持FLOAT，FLOAT16，BFLOAT16。
  - [object Object]Atlas 推理系列产品[object Object]、[object Object]Atlas 训练系列产品[object Object]：数据类型支持FLOAT，FLOAT16。
- cubeMathType(int8_t, 计算输入)：用于判断Cube单元应该使用哪种计算逻辑进行运算，数据类型为INT8，注意：如果输入的数据类型存在互推导关系，该参数默认对互推导后的数据类型进行处理。支持的枚举值如下：
  - 0:KEEP_DTYPE，保持输入的数据类型进行计算。
    - [object Object]Atlas 训练系列产品[object Object]、[object Object]Atlas 推理系列产品[object Object]：当输入是FLOAT，暂不支持该选项。
  - 1:ALLOW_FP32_DOWN_PRECISION，允许将输入数据降精度计算。
    - [object Object]Atlas 训练系列产品[object Object]、[object Object]Atlas 推理系列产品[object Object]：当输入是FLOAT，允许转换为FLOAT16计算。当输入为其他数据类型时不做处理。
    - [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品[object Object]：当输入是FLOAT，允许转换为HFLOAT32计算。当输入为其他数据类型时不做处理。
  - 2:USE_FP16，允许转换为数据类型FLOAT16进行计算。
    - [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品[object Object]：当输入是BFLOAT16时不支持该选项。
  - 3:USE_HF32，允许转换为数据类型HFLOAT32计算。
    - [object Object]Atlas 训练系列产品[object Object]、 [object Object]Atlas 推理系列产品[object Object]：暂不支持该选项。
    - [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品[object Object]：当输入是FLOAT，转换为HFLOAT32计算。当输入为其他数据类型时不做处理。
- workspaceSize(uint64_t *，出参)：返回需要在Device侧申请的workspace大小。
- executor(aclOpExecutor **，出参)：返回op执行器，包含了算子计算流程。
返回值：

aclnnStatus：返回状态码，具体参见。

[object Object]

aclnnConvTbc

参数说明：
- workspace(void *, 入参)：在Device侧申请的workspace内存地址。
- workspaceSize(uint64_t, 入参)：在Device侧申请的workspace大小，由第一段接口aclnnConvTbcGetWorkspaceSize获取。
- executor(aclOpExecutor *, 入参)：op执行器，包含了算子计算流程。
- stream(aclrtStream, 入参)：指定执行任务的AscendCL Stream流。
返回值： aclnnStatus：返回状态码，具体参见。

约束与限制

无

调用示例

示例代码如下，仅供参考，具体编译和执行过程请参考。

[object Object]