torch-mlir/lib/Dialect/Torch/Transforms
Qiang Fu f7c7bb800c
Add non-default dtype support for a few elementwise math ops. (#687)
* fix type inference
* fix Torch2Linalg conversion
* add test cases
2022-03-23 13:35:43 -07:00
..
AdjustCallingConventions.cpp Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00
CMakeLists.txt Introduce new shape library design. 2022-03-15 12:41:58 -07:00
DecomposeComplexOps.cpp [LINALG] Add decomposition of `aten.dropout` op 2022-03-22 13:14:49 +05:30
DropShapeCalculations.cpp Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00
GlobalizeObjectGraph.cpp Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00
InlineGlobalSlots.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
MaximizeValueSemantics.cpp Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00
PassDetail.h Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
Passes.cpp Introduce new shape library design. 2022-03-15 12:41:58 -07:00
PrepareForGlobalizeObjectGraph.cpp Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00
ReduceOpVariants.cpp [MLIR][TORCH] Add value tensor variant to aten::copy_ op 2022-03-21 19:12:37 +05:30
RefinePublicReturn.cpp Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00
RefineTypes.cpp Add non-default dtype support for a few elementwise math ops. (#687) 2022-03-23 13:35:43 -07:00
ReifyShapeCalculations.cpp Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00
ShapeLibrary.cpp Add non-default dtype support for a few elementwise math ops. (#687) 2022-03-23 13:35:43 -07:00
SimplifyShapeCalculations.cpp Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00