torch-mlir/lib/Conversion/TorchToLinalg
dan 7750d2173a add argmax lowering
Add argmax lowering from torch to linalg
2021-10-13 14:31:16 -04:00
..
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
TorchToLinalg.cpp add argmax lowering 2021-10-13 14:31:16 -04:00