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

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