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 Conv3DBackpropInput objects. Resources are divided based on the tiling parameters. For details about the tiling parameters, see Conv3DBackpropInput Tiling API.

Prototype

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

Parameters

Table 1 API parameters

Parameter

Input/Output

Description

tiling

Input

Tiling parameter of the Conv3DBackpropInput object. For details about the definition of the TConv3DBackpropInputTiling structure, see TConv3DBackpropInputTiling Structure.

Tiling parameters can be obtained through GetTiling on the host and passed to the kernel 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 Conv3DBackpropInputTilingData* tilingData;
//... Create a Conv3DBackpropInput object and call the init API after initializing tilingData.
ConvBackpropApi::Conv3DBackpropInput<weightDxType, inputSizeDxType, gradOutputDxType, gradInputDxType> gradInput_;
gradInput_.Init(&(tilingData->conv3DDxTiling));