[stablehlo] Add batching dims to stablehlo.gather
and stablehlo.scatter
#5405
Job | Run time |
---|---|
9s | |
26s | |
15s | |
6s | |
5s | |
8s | |
1m 9s |
stablehlo.gather
and stablehlo.scatter
#5405
Job | Run time |
---|---|
9s | |
26s | |
15s | |
6s | |
5s | |
8s | |
1m 9s |