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