SetSingleWeightShape

Function

Sets the shape of the weight matrix on a single core.

Prototype

void SetSingleWeightShape(int64_t singleCi, int64_t singleKd, int64_t singleKh, int64_t singleKw)

Parameters

Parameter	Input/Output	Description
singleCi	Input	Size of the input channel on a single core.
singleKd	Input	Size of the Weight D dimension on a single core.
singleKh	Input	Size of the Weight H dimension on a single core.
singleKw	Input	Size of the Weight W dimension on a single core.

Returns

None

Restrictions

None

Example

// Instantiate the Conv3D API.
auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo());
Conv3dTilingApi::Conv3dTiling conv3dApiTiling(ascendcPlatform );
conv3dApiTiling.SetSingleWeightShape(singleCi, singleKd, singleKh, singleKw);

Parent topic: Conv3D Tiling APIs