SetWeight

Applicability

Product	Supported
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas 200I/500 A2 inference products	x
Atlas inference product's AI Core	x
Atlas inference product's Vector Core	x
Atlas training products	x

Function

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

Example

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

Parent topic: Conv3DBackpropInput Kernel API