llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Conversion/TorchToLinalg
History
Vivek Khandelwal 769f3a8870 [MLIR][TORCH] Add E2E support for max_pool2d_with_indices op 
This commit adds lowering of `max_pool2d_with_indices` op.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
 		2022-04-18 21:05:19 +05:30
..
CMakeLists.txt Split up TorchToLinalg.cpp 2022-03-14 10:19:41 -07:00
DataMovement.cpp [MLIR][TORCH] Add support for same input and output shapes for view op 2022-03-25 22:26:10 +05:30
IndirectDataMovement.cpp Split up TorchToLinalg.cpp 2022-03-14 10:19:41 -07:00
Linear.cpp [MLIR][TORCH] Add E2E support for max_pool2d_with_indices op 2022-04-18 21:05:19 +05:30
Pooling.cpp [MLIR][TORCH] Add E2E support for max_pool2d_with_indices op 2022-04-18 21:05:19 +05:30
PopulatePatterns.h Split up TorchToLinalg.cpp 2022-03-14 10:19:41 -07:00
Random.cpp Improve "pseudo" op terminology. 2022-03-15 17:57:52 -07:00
Reduction.cpp Split up TorchToLinalg.cpp 2022-03-14 10:19:41 -07:00
TensorConstructors.cpp [LINALG] Add torch.layout information 2022-04-07 20:47:49 +05:30
TensorScalarInterop.cpp Split up TorchToLinalg.cpp 2022-03-14 10:19:41 -07:00
TorchToLinalg.cpp Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00
Uncategorized.cpp Add shape transfer function and lowering to linalg for aten.neg (#759) 2022-04-15 11:11:22 -07:00
Utils.cpp [MLIR][TORCH] Add E2E support for max_pool2d_with_indices op 2022-04-18 21:05:19 +05:30
Utils.h [MLIR][TORCH] Add E2E support for max_pool2d_with_indices op 2022-04-18 21:05:19 +05:30