Reimplement eye op on TF to support tensors as inputs.
#1782
| Job | Run time |
|---|---|
| 4s | |
| 4s |
eye op on TF to support tensors as inputs.
#1782
| Job | Run time |
|---|---|
| 4s | |
| 4s |