[object Object]

[object Object][object Object]undefined
[object Object]
  • Description: Retrieves a set of values from weight based on indices, then aggregates them using the specified mode (max, sum, or mean) according to the offsets defined by offsets. Other parameters provide finer control over the computation process.

  • Shape derivation: Assume:

    [object Object]
    • When mode is sum:

      [object Object]
    • When mode is mean:

      [object Object]
    • When mode is max:

      [object Object]
[object Object]

Each operator has calls. First, aclnnEmbeddingBagGetWorkspaceSize is called to obtain the workspace size required for computation and the executor that contains the operator computation flow. Then, aclnnEmbeddingBag is called to perform computation.

[object Object]
[object Object]
[object Object]
  • Parameters

    [object Object]
    • [object Object]Atlas training series products[object Object] and [object Object]Atlas inference series products[object Object]: The data type cannot be BFLOAT16.
  • Returns

    aclnnStatus: status code. For details, see .

    The first-phase API implements input parameter verification. The following errors may be thrown.

    [object Object]
[object Object]
  • Parameters

    [object Object]
  • Returns

    aclnnStatus: status code. For details, see .

[object Object]
  • Deterministic computation:
    • aclnnEmbeddingBag defaults to a deterministic implementation.

[object Object] and [object Object] support only [object Object].

[object Object]

The following example is for reference only. For details, see .

[object Object]