SetWeightShape

Function

Sets the shape of the weight matrix.

Prototype

void SetWeightShape(int64_t cout, int64_t cin, int64_t d, int64_t h, int64_t w)

Parameters

**Table 1** Parameters
Parameter	Input/Output	Description
cout	Input	Sets the channel value of GradOutput.
cin	Input	Sets the channel value of Input.
d	Input	Sets the depth value of Weight.
h	Input	Sets the height value of Weight.
w	Input	Sets the width value of Weight.

Returns

None

Restrictions

None

Example

optiling::Conv3DBackpropFilterTilingData tilingData;
auto ascendcPlatform = platform_ascendc::PlatformAscendCManager::GetInstance();
ConvBackpropApi::Conv3dBpFilterTiling conv3dBpDwTiling(*ascendcPlatform);
conv3dBpDwTiling.SetWeightShape(cout, cin, d, h, w);

Parent topic: Conv3DBackpropFilter Tiling API