SetInput

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 feature matrix Input.

Prototype

__aicore__ inline void SetInput(const AscendC::GlobalTensor<SrcT> &input)

Parameters

**Table 1** API parameters
Parameter	Input/Output	Description
input	Input	Start address of the input in the global memory. The type is GlobalTensor. The feature matrix Input supports the following SrcT data types: half and bfloat16_t.

Returns

None

Restrictions

None

Example

const Conv3DBackpropFilterTilingData* tilingData;
//... Initialize tilingData.
ConvBackpropApi::Conv3DBackpropFilter <inputType, weightSizeType, gradOutputType, gradWeightType > gradWeight_;
gradWeight_.Init(&(tilingData->dwTiling));
// Set the Input.
gradWeight_.SetInput(inputGm_[offsetB_]);
...

Parent topic: Conv3DBackpropFilter Kernel API