torch-mlir/lib/Dialect/Torch/Transforms
Gaurav Shukla 969785d1b6 [LINALG] Add E2E support for `aten.where.[Scalar|ScalarSelf|ScalarOther]` ops
This commit decomposes different variants of `aten.where.*` op into
`aten.where.Self` op. It covers `aten.where.Scalar`,
`aten.where.ScalarSelf` and `aten.where.ScalarOther` ops.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
2022-03-30 20:36:48 +05:30
..
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 E2E support for `aten.where.[Scalar|ScalarSelf|ScalarOther]` ops 2022-03-30 20:36:48 +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 This commit decomposes aten._reshape_alias op into aten.view op. (#690) 2022-03-28 23:54:28 -05: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 [LINALG] Add E2E support for `aten.zero_` op 2022-03-25 12:46:50 +05:30
RefinePublicReturn.cpp Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00
RefineTypes.cpp [LINALG] Add E2E support for `aten.where.[Scalar|ScalarSelf|ScalarOther]` ops 2022-03-30 20:36:48 +05:30
ReifyShapeCalculations.cpp Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00
ShapeLibrary.cpp [LINALG] Add E2E support for `aten.where.[Scalar|ScalarSelf|ScalarOther]` ops 2022-03-30 20:36:48 +05:30
SimplifyShapeCalculations.cpp Improve list index normalization SimplifyShapeCalculations. (#710) 2022-03-29 22:21:47 +02:00