torch-mlir/lib/Conversion/TorchToStablehlo
Jiawei Wu c93c6970e8
[stablehlo] add dtype conversion when converting AtenScalarImplicitOp (#2439)
2023-09-06 01:57:15 +08:00
..
Basic.cpp [stablehlo] add dtype conversion when converting AtenScalarImplicitOp (#2439) 2023-09-06 01:57:15 +08:00
CMakeLists.txt [StableHLO] Support for slice_scatter (#1960) 2023-03-22 13:41:04 -07:00
GatherScatter.cpp [MLIR][TORCH] Add StableHLO lowering for embedding_bag.padding_idx op 2023-09-05 21:32:23 +05:30
Linear.cpp [Stablehlo] fix promoteType() when input doesn't have DefiningOp (#2262) 2023-06-26 00:04:17 +08:00
Pooling.cpp [Stablehlo]Add support for AvgPool1dOp (#2268) 2023-07-25 14:09:53 +08:00
PopulatePatterns.h [StableHLO] Support for slice_scatter (#1960) 2023-03-22 13:41:04 -07:00
Reduction.cpp [MLIR][TORCH] add E2E support for aten.min op (#2422) 2023-08-29 12:12:41 -05:00
StablehloLegalizeUtils.cpp [Stablehlo] fix promoteType() when input doesn't have DefiningOp (#2262) 2023-06-26 00:04:17 +08:00
TorchToStablehlo.cpp [Stablehlo] support dynamic shape when convert aten.fill.Scalar (#2349) 2023-07-27 18:35:25 +08:00
ViewLike.cpp [MLIR][TORCH] Patch up Ops and their lowerings to deal with +ve `dim` 2023-04-14 13:12:56 +05:30