torch-mlir/lib/Conversion/TorchToStablehlo
Yuanqiang Liu 6f7d9e83df
[Stablehlo] add e2e test for aten.batch_norm (#2129)
2023-05-17 09:04:40 -07:00
..
Basic.cpp [Stablehlo] add e2e test for aten.batch_norm (#2129) 2023-05-17 09:04:40 -07:00
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] fix reduce max init_value with -inf (#2064) 2023-05-06 12:05:51 -07:00
PopulatePatterns.h [StableHLO] Support for slice_scatter (#1960) 2023-03-22 13:41:04 -07:00
Reduction.cpp [Stablehlo] fix reduce max init_value with -inf (#2064) 2023-05-06 12:05:51 -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