llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Dialect/Torch/Transforms
History
Yi Zhang a459e09ab7 E2e support for aten.softmax.int and aten.embedding 
- Added a DecomposeComplexOps pass to decompose complex torchOps.
- Refactored `visitAtenArgmaxOp` and `visitAtenAnyDimOp` to
`visitReductionAlongDimIntOp`.
- Moved some helper functions into
torch-mlir/Dialect/Torch/Utils/Utils.h to be shared by multiple files.
- Added support for f64 tensor as argument and return types.
 		2021-10-18 17:57:45 -04:00
..
AdjustCallingConventions.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00
DecomposeComplexOps.cpp E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04: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 E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00
PassDetail.h Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
Passes.cpp E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04: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 E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00