SetInput

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

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

Examples

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 APIs