Init

Applicability

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

Function Usage

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

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 calling sequence is as follows:
1
2
3
4
5
Init(...);
...
Iterate(...);
GetTensorC(...);
End();

Examples

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));