torch-mlir/lib/Dialect
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
..
Torch [MLIR][TORCH] Add lowering for `aten._softmax` when `half_to_float=True` 2022-11-28 22:32:00 +05:30
TorchConversion llvm: update tag to e864ac6945 (#1600) 2022-11-16 14:40:36 -08:00
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00