torch-mlir/lib/Conversion/TorchToLinalg
Jerin Philip 9cb5d38cd1
[MLIR][TORCH] Add E2E `torch.aten.prod_dim_int` (#2423)
Uses the existing reduction codepath, adding modifications or branches
required alongside for prod.
2023-09-05 13:38:51 -07:00
..
CMakeLists.txt Re-enable custom op support 2022-08-16 22:49:08 +05:30
DataMovement.cpp [MLIR][TORCH] Add E2E support for view_as_real op (#2419) 2023-09-01 21:12:01 -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 [build] Update llvm tag to a3f2751f (#2397) 2023-08-15 09:53:28 -07: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 [MLIR][TORCH] Add E2E `torch.aten.prod_dim_int` (#2423) 2023-09-05 13:38:51 -07:00
TensorConstructors.cpp [build] Update llvm tag to a3f2751f (#2397) 2023-08-15 09:53:28 -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 [MLIR][TORCH] Add e2e support for aten.pow.Scalar op 2023-08-31 21:43:24 +05:30
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