SetWeight

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

__aicore__ inline void SetWeight(const AscendC::GlobalTensor<WeightT>& weight)

Parameter	Input/Output	Description
weight	Input	Address of Weight in the Global Memory. The type is GlobalTensor. The Weight matrix supports the following data types: half and bfloat16_t.

None

None

GlobalTensor<half> weightGm;
weightGm.SetGlobalBuffer(reinterpret_cast<__gm__ half *>(weight));
conv3dApi.SetWeight(weightGm);

Parent topic: Conv3D Kernel APIs