xla
3f62f05d - Add _argmin and _argmax to ATen XLA tensor

Commit

6 years ago

Add _argmin and _argmax to ATen XLA tensor They're just wrappers for argmin and argmax.

References

Author

asuhan

Committer

asuhan

Parents