Init

Applicability

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

Function

Initializes tiling data in Conv3DBackpropFilter objects. Resources are divided based on the tiling parameters. For details about the tiling parameters, see Conv3DBackpropFilter Tiling API.

Prototype

1
__aicore__ inline void Init(const TConv3DBpFilterTiling *__restrict tiling)

Parameters

Table 1 API parameters

Parameter

Input/Output

Description

tiling

Input

Tiling parameters of the Conv3DBackpropFilter object. For details about the Conv3DBackpropFilterTilingData structure, see TConv3DBpFilterTiling Structure.

Tiling parameters can be obtained through the GetTiling API on the Host side and passed to the kernel side for use.

Returns

None

Restrictions

The Init API must be called before the Iterate, GetTensorC, and End APIs, and can be called only once. The call sequence is as follows:
1
2
3
4
5
Init(...);
...
Iterate(...);
GetTensorC(...);
End();

Example

1
2
3
4
const Conv3DBackpropFilterTilingData* tilingData;
//... Create a Conv3DBackpropFilter object and call the init API after initializing tilingData.
ConvBackpropApi::Conv3DBackpropFilter <inputType, weightSizeType, gradOutputType, gradWeightType > gradWeight_;
gradWeight_.Init(&(tilingData->dwTiling));