[NDTensors] Scalar diag calls expose to skip scalar indexing
#4558
| Job | Run time |
|---|---|
| 2m 47s | |
| 2m 47s |
expose to skip scalar indexing
#4558
| Job | Run time |
|---|---|
| 2m 47s | |
| 2m 47s |