SetInput

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

Sets the feature matrix Input.

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

Parameter	Input/Output	Description
input	Input	Start address of Input in the Global Memory. The type is GlobalTensor. The feature matrix Input 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