[NDTensors] Scalar diag calls expose to skip scalar indexing
#2780
| Job | Run time |
|---|---|
| 58s | |
| 58s |
expose to skip scalar indexing
#2780
| Job | Run time |
|---|---|
| 58s | |
| 58s |