torch-mlir/lib/Conversion/TorchToStablehlo
Jiawei Wu 60bad54f27
[Torch Dialect] replace none-index in aten.Index.Tensor's param by manually generating it (#2344)
* [Torch Dialect] replace none-index in aten.Index.Tensor's  param by manually generating it
Co-authored-by: Jiawei Wu <wujiawei.aml@bytedance.com>
Co-authored-by: Jianzhe Xiao <jianzhe.xiao@bytedance.com>

* minor typo fix

* add new failed e2e tests for ltc

* fix typo

* Address comments

* Add more e2e tests

* add failed e2e tests for LTC

* address comments

* remove decomposition for AtenIndexTensorHackedTwinOp
2023-08-15 19:36:08 +08:00
..
Basic.cpp Fix "unused variable 'outType'" warning (#2378) 2023-08-05 07:41:01 +08:00
CMakeLists.txt [StableHLO] Support for slice_scatter (#1960) 2023-03-22 13:41:04 -07:00
GatherScatter.cpp [Torch Dialect] replace none-index in aten.Index.Tensor's param by manually generating it (#2344) 2023-08-15 19:36:08 +08:00
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 Bump LLVM (#2176) 2023-06-13 16:17:23 +02: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