torch-mlir/lib/Dialect/Torch/Transforms
Abhishek Varma bb259f918a [MLIR][TORCH] Add lowering for `aten._softmax` when `half_to_float=True`
-- This commit adds decompose logic for `aten._softmax` when
   `half_to_float` is `True`.
-- An e2e test case will be added once support for half to float conversion for
   `aten._softmax` is added upstream.

Signed-off-by: Abhishek Varma <abhishek@nod-labs.com>
2022-11-28 22:32:00 +05:30
..
AdjustCallingConventions.cpp Add better error message for single-tensor tuple returns. 2022-10-25 12:48:55 -04:00
CMakeLists.txt Iteratively run the main simplification pipeline. 2022-08-17 14:54:33 -07:00
DecomposeComplexOps.cpp [MLIR][TORCH] Add lowering for `aten._softmax` when `half_to_float=True` 2022-11-28 22:32:00 +05:30
DropShapeCalculations.cpp torch: allow torch dialect ops after running drop-shape pass (#979) 2022-06-25 07:27:47 -07:00
EraseModuleInitializer.cpp Iteratively run the main simplification pipeline. 2022-08-17 14:54:33 -07:00
GlobalizeObjectGraph.cpp llvm: update tag to 061e0189 (#1180) 2022-08-08 20:17:35 -07:00
InlineGlobalSlots.cpp Revert "build: update llvm tag to 147fe9de" 2022-11-25 12:41:56 +05:30
LowerToBackendContract.cpp Fix Base Lazy Backend Type Conversion (#1412) 2022-10-04 15:53:28 -07:00
MaximizeValueSemantics.cpp [tosa] Add support for some cases of aten.broadcast_to op (#1429) 2022-09-29 09:40:56 -07:00
PassDetail.h llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
Passes.cpp Fix Base Lazy Backend Type Conversion (#1412) 2022-10-04 15:53:28 -07:00
PrepareForGlobalizeObjectGraph.cpp llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
ReduceOpVariants.cpp Remove all but one of valsem ops + move fill.Scalar to elementwise (#1531) 2022-10-28 15:06:11 +00:00
RefinePublicReturn.cpp Add support for returning more than one copy of the same tensor (#1228) 2022-08-18 22:41:45 +00:00
RefineTypes.cpp [MLIR][TORCH] Fix refine types crash 2022-11-23 15:17:37 +05:30
ReifyShapeCalculations.cpp llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
ShapeLibrary.cpp Add more mhlo basic converters (#1628) 2022-11-24 14:28:34 +08:00
SimplifyShapeCalculations.cpp [MLIR][TORCH] Add support for multiple indexing tensors for aten.index.Tensor (#1097) 2022-07-28 19:00:02 -04:00