SetWeight

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 Weight for convolution backpropagation computation.

Prototype

__aicore__ inline void SetWeight(const AscendC::GlobalTensor<SrcT> &weight)

Parameters

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

Returns

None

Restrictions

None

Examples

const Conv3DBackpropInputTilingData* tilingData;
//... Initialize tilingData.
ConvBackpropApi::Conv3DBackpropInput<weightDxType, inputSizeDxType, gradOutputDxType, gradInputDxType> gradInput_;
//... Set other parameters.
gradInput_.SetWeight(weightGm_[offsetB_]);

Parent topic: Conv3DBackpropInput Kernel APIs