[NDTensors] Scalar diag calls expose to skip scalar indexing
#2786
| Job | Run time |
|---|---|
| 1m 4s | |
| 1m 4s |
expose to skip scalar indexing
#2786
| Job | Run time |
|---|---|
| 1m 4s | |
| 1m 4s |