SetInput

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

Sets the Input feature matrix.

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

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

None

None

GlobalTensor<half> inputGm;
inputGm.SetGlobalBuffer(reinterpret_cast<__gm__ half *>(input));
conv3dApi.SetInput(inputGm);

Parent topic: Conv3D Kernel APIs