[stablehlo] Add batching dims to `stablehlo.gather` and `stablehlo.scatter`#2259

Merged

ghpvnist merged 50 commits intoopenxla:mainfrom tomnatan30:gather-scatter-batch-impl

May 15, 2024

+4,872-618

Commits on Mar 12, 2024

[RFC] Add batching dims to stablehlo.gather and stable.scatter specification
tomnatan30
committed

Commits on Mar 21, 2024

Commits on Mar 22, 2024

Commits on Apr 23, 2024

fix arrows in image
tomnatan30
committed

Commits on Apr 25, 2024

Merge branch 'openxla:main' into gather-scatter-batch-spec
tomnatan30
authored

Commits on May 1, 2024

Commits on May 2, 2024

Commits on May 7, 2024

Commits on May 8, 2024

Commits on May 9, 2024

Update table formatting and fix typos (openxla#2300 )

ghpvnist
authored and
tomnatan30
committed
dynamic_reshape op spec (openxla#2284 )

abhigunj
authored and
tomnatan30
committed
Change uses of deprecated method a.cast to cast<>(a) (openxla#2301 )

abhigunj
authored and
tomnatan30
committed
Remove dynamic_{iota,reshape} from list of ops to spec (openxla#2303 )

mlevesquedion
authored and
tomnatan30
committed
dynamic_iota op : match ODS description with the spec (openxla#2305 )

abhigunj
authored and
tomnatan30
committed
Merge branch 'main' into gather-scatter-batch-impl
tomnatan30
authored