torch-mlir/lib/Conversion/TorchToLinalg
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
..
CMakeLists.txt Re-enable custom op support 2022-08-16 22:49:08 +05:30
DataMovement.cpp [Torch-to-Linalg] Add dynamic dimension support for BroadcastTo op (#2174) 2023-07-07 10:01:51 -07:00
IndirectDataMovement.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] add conversion for AtenFlipOp (#2163) 2023-06-15 10:27:34 +08:00
Pooling.cpp [Stablehlo]Add support for AvgPool1dOp (#2268) 2023-07-25 14:09:53 +08:00
PopulatePatterns.h Re-enable custom op support 2022-08-16 22:49:08 +05:30
Random.cpp build: update llvm tag to 798fa4b4 (#1684) 2022-12-07 12:20:41 -08:00
Reduction.cpp [Linalg] fix lowering reduce max with -inf (#2097) 2023-05-08 09:17:49 -07:00
TensorConstructors.cpp Fix creation of empty tensor in decomposition for randn ops (#2043) 2023-04-19 08:25:39 -07:00
TensorScalarInterop.cpp build: update llvm tag to 798fa4b4 (#1684) 2022-12-07 12:20:41 -08:00
TorchToLinalg.cpp Add complex types support with basic complex ops. 2023-05-11 21:29:07 +05:30
Uncategorized.cpp lib/Conversion/TorchToTosa/TorchToTosa.cpp: Fix legalization of comparions where the input type is bool (#2304) 2023-07-17 09:49:04 +02:00
Utils.cpp [Torch-to-Linalg] Add dynamic dimension support for BroadcastTo op (#2174) 2023-07-07 10:01:51 -07:00
Utils.h [Torch-to-Linalg] Add dynamic dimension support for BroadcastTo op (#2174) 2023-07-07 10:01:51 -07:00