SetWeight

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

__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 Weight 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