Request the decomposition for gatherElements, scatterElements and scatterND #767
Description
| Operator | Notes | TFLite |
|---|---|---|
| gatherElements | Gathers values from the input using indices along the given axis. Related: mil.ops.defs.iOS15.scatter_gather.gather_along_axis, tf.experimental.numpy.take_along_axis?, DML_GATHER_ELEMENTS Decomposition: NA Data Types: input (*), indices (int32, uint32, int64) |
|
| scatterElements | Corollary for pair-completeness with gatherElements. Related: mil.ops.defs.iOS17.scatter_gather.scatter_along_axis, TF=?, DML_SCATTER_ELEMENTS Decomposition: NA Data Types: input (*), updates (same as input), indices (int32, uint32, int64) |
|
| scatterND | Scatter values using multidimensional indices. This is also useful for improving performance of ferrying partial MLBuffer transputs between iterations (transformer key-value reuse). Related: tf.scatter_nd, mil.ops.defs.iOS15.scatter_gather.scatter_nd, DML_SCATTER_ND Decomposition: NA Data Types: input (*), updates (same as input), indices (int32, uint32, int64) |
tf.scatter_nd only scatter zero-initialized tensor with the third argument that only specifies shape not value like webnn's input |