Bringing Tensor::gather()
behavior closer to torch.gather()
#6037
Job | Run time |
---|---|
10s | |
10s | |
10s | |
1m 14s | |
1m 14s | |
1m 14s | |
1m 52s | |
1m 52s | |
1m 52s | |
47m 11s | |
20m 6s | |
45m 0s | |
2m 49s | |
2m 49s | |
2m 49s | |
8m 54s | |
8m 54s | |
8m 54s | |
1m 34s | |
1m 34s | |
1m 34s | |
3m 28s | |
3m 28s | |
3m 28s | |
2h 52m 20s |