End

产品支持情况

产品

是否支持

Ascend 950PR/Ascend 950DT

x

Atlas A3 训练系列产品/Atlas A3 推理系列产品

√

Atlas A2 训练系列产品/Atlas A2 推理系列产品

√

Atlas 200I/500 A2 推理产品

x

Atlas 推理系列产品AI Core

x

Atlas 推理系列产品Vector Core

x

Atlas 训练系列产品

x

功能说明

在Conv3DBackpropInput卷积反向计算完成后,必须调用一次End,以清除EventID并释放内部申请的临时内存。

函数原型

1
__aicore__ inline void End()

参数说明

无

返回值说明

无

约束说明

End接口必须在Iterate和GetTensorC接口后调用。

调用示例

1
2
3
4
5
const Conv3DBackpropInputTilingData* tilingData;
// ...初始化tilingData
ConvBackpropApi::Conv3DBackpropInput<weightDxType, inputSizeDxType, gradOutputDxType, gradInputDxType> gradInput_;
// ...其它调用
gradInput_.End();