torch-mlir/lib/Dialect/Torch/Transforms
dan 7750d2173a add argmax lowering
Add argmax lowering from torch to linalg
2021-10-13 14:31:16 -04:00
..
AdjustCallingConventions.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
GlobalizeObjectGraph.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
InlineGlobalSlots.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
MaximizeValueSemantics.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
PassDetail.h Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
Passes.cpp Add a basic TOSA E2E backend. 2021-10-08 09:59:45 -07:00
PrepareForGlobalizeObjectGraph.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
ReduceOpVariants.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
RefinePublicReturn.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
RefineTypes.cpp add argmax lowering 2021-10-13 14:31:16 -04:00