Init
Applicability
Product |
Supported |
|---|---|
√ |
|
√ |
|
x |
|
x |
|
x |
|
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
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
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)); |