torch-mlir/lib/Conversion/TorchToLinalg
JianzheXiao 31ef08b63d
[Stablehlo]Add support for AvgPool1dOp (#2268)
* Add support for AvgPool1d

* Update AbstractInterpLibrary

* support avgpool1d in linalg

* refactored code

* fix nit problem
2023-07-25 14:09:53 +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 [MLIR][TORCH] Add support for negative index values for index.Tensor op (#2233) 2023-06-16 14:21:04 -05: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