torch-mlir/lib/Conversion/TorchToStablehlo
Abhishek Varma 318fe13468 [MLIR][TORCH] Patch up Ops and their lowerings to deal with +ve `dim`
-- In Python we have the concept of negative dimension indexing.
-- We would want to normalize such dimensions to be +ve and within the
   expected range instead.
-- This commit takes care of a few remaining set of Ops and their
   lowerings by applying `toPositiveDim` and `isValidDim` to the
   extracted integer `dim` value.

Signed-off-by: Abhishek Varma <abhishek@nod-labs.com>
2023-04-14 13:12:56 +05:30
..
Basic.cpp [MLIR][TORCH] Patch up Ops and their lowerings to deal with +ve `dim` 2023-04-14 13:12:56 +05:30
CMakeLists.txt [StableHLO] Support for slice_scatter (#1960) 2023-03-22 13:41:04 -07:00
GatherScatter.cpp [MLIR][TORCH] Patch up Ops and their lowerings to deal with +ve `dim` 2023-04-14 13:12:56 +05:30
Linear.cpp [StableHlo] Fix transposed convolution conversion (#2026) 2023-04-13 11:24:39 -07:00
Pooling.cpp [STABLEHLO] Move utils.h to include/ (#1974) 2023-03-27 21:16:21 -07:00
PopulatePatterns.h [StableHLO] Support for slice_scatter (#1960) 2023-03-22 13:41:04 -07:00
Reduction.cpp [STABLEHLO] Move utils.h to include/ (#1974) 2023-03-27 21:16:21 -07:00
StablehloLegalizeUtils.cpp [STABLEHLO] Move utils.h to include/ (#1974) 2023-03-27 21:16:21 -07:00
TorchToStablehlo.cpp [StableHLO] Support for slice_scatter (#1960) 2023-03-22 13:41:04 -07:00
ViewLike.cpp [MLIR][TORCH] Patch up Ops and their lowerings to deal with +ve `dim` 2023-04-14 13:12:56 +05:30