torch-mlir/lib/Conversion/TorchToLinalg
Quinn Dawkins 53bf09ceef
Fix iterator types for embedding bag sum mode (#1371)
2022-09-23 13:13:47 -04:00
..
CMakeLists.txt Re-enable custom op support 2022-08-16 22:49:08 +05:30
DataMovement.cpp Revert "Implement Non-Expand/Collapse Functionality for Aten.View (#1309)" (#1347) 2022-09-07 01:38:11 -04:00
IndirectDataMovement.cpp Fix iterator types for embedding bag sum mode (#1371) 2022-09-23 13:13:47 -04:00
Linear.cpp [TORCH] Add decomposition of `aten.linear` op 2022-09-07 16:58:27 +05:30
Pooling.cpp [MLIR][TORCH] Add decomposition of aten.adaptive_avg_pool2d op 2022-05-27 07:56:37 +05:30
PopulatePatterns.h Re-enable custom op support 2022-08-16 22:49:08 +05:30
Random.cpp Improve "pseudo" op terminology. 2022-03-15 17:57:52 -07:00
Reduction.cpp [MLIR][MHLO] Add AtenFrobeniusNormDimOp and add its conversion pattern to MHLO and linalg (#1306) 2022-09-08 10:15:36 +08:00
TensorConstructors.cpp [LINALG] Add support for preserve memory format in aten_empty_like op. 2022-05-10 09:37:55 +05:30
TensorScalarInterop.cpp [MLIR][TORCH] Add E2E support for ScalarImplicit, Int.Scalar op 2022-05-10 22:40:49 +05:30
TorchToLinalg.cpp Re-enable custom op support 2022-08-16 22:49:08 +05:30
Uncategorized.cpp [MLIR][TORCH] Add E2E support for aten.bitwise_not op 2022-09-08 17:52:12 +05:30
Utils.cpp Add transposed case for at::convolution (#917) 2022-08-24 12:19:35 -04:00
Utils.h Add transposed case for at::convolution (#917) 2022-08-24 12:19:35 -04:00