TilingInputsDataDependency
Description
Specifies the indexth input tensor on which the tiling computation depends and the execution position of the tiling computation.
Prototype
OpImplRegisterV2 &TilingInputsDataDependency(std::initializer_list<int32_t> inputs)
OpImplRegisterV2 &TilingInputsDataDependency(std::initializer_list<int32_t> inputs, std::initializer_list<TilingPlacement> placements)
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
inputs |
Input |
List of input indexes on which operator tiling computation depends. For example, inputs={0, 3} indicates that the operator tiling computation depends on the tensor values of input 0 and input 3. |
placements |
Input |
Position where the operator tiling can be executed. 0 indicates that the operator tiling can be executed on the host, and 1 indicates that the operator tiling can be executed on the AI CPU. If this parameter is not included, the operator tiling can be executed only on the host. |
Returns
OpImplRegisterV2 object of the operator, where the indexth input tensor on which the tiling computation depends and the execution position are registered.
Restrictions
None