SetGradOutput

Applicability

Product	Supported
Atlas 350 Accelerator Card	x
Atlas A3 training product/Atlas A3 inference product	√
Atlas A2 training product/Atlas A2 inference product	√
Atlas 200I/500 A2 inference product	x
Atlas inference product AI Core	x
Atlas inference product Vector Core	x
Atlas training product	x

Function Usage

Sets the input matrix GradOutput for convolution backpropagation computation.

Prototype

__aicore__ inline void SetGradOutput(const AscendC::GlobalTensor<SrcT> &gradOutput)

Parameters

**Table 1** API parameters
Parameter	Input/Output	Description
gradOutput	Input	Start address of the GradOutput matrix in the global memory. The type is GlobalTensor. SrcT indicates the data type of the GradOutput matrix. Currently, the supported data types are half and bfloat16_t.

Returns

None

Restrictions

None

Examples

ConvBackpropApi::Conv3DBackpropInput<weightDxType, inputSizeDxType, gradOutputDxType, gradInputDxType> gradInput_;
// Set the address of the GlobalTensor in GradOutput.
GlobalTensor<gradOutputType> gradOutputGm_;
gradOutputGm_.SetGlobalBuffer((__gm__ gradOutputType *)gradOutput);
gradInput_.SetGradOutput(gradOutputGm_);

Parent topic: Conv3DBackpropInput Kernel APIs