Vivek Khandelwal 4b11284440 [MLIR][TORCH] Add E2E support for aten.avg_pool2d op ... This commit adds lowering of `aten.avg_pool2d` op. Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>		2022-05-02 12:31:44 +05:30
..
CMakeLists.txt	Split up TorchToLinalg.cpp	2022-03-14 10:19:41 -07:00
DataMovement.cpp	[LINALG] Bug fix i64 vs i32 type comparison.	2022-04-22 08:09:58 +05:30
IndirectDataMovement.cpp	[LINALG] Bug fix i64 vs i32 type comparison.	2022-04-22 08:09:58 +05:30
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 aten.avg_pool2d op	2022-05-02 12:31:44 +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] Bug fix i64 vs i32 type comparison.	2022-04-22 08:09:58 +05:30
TensorScalarInterop.cpp	[LINALG] Bug fix i64 vs i32 type comparison.	2022-04-22 08:09:58 +05:30
TorchToLinalg.cpp	llvm: bump tag to e1318078 (#781)	2022-04-26 12:27:51 -07: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	[LINALG] Bug fix i64 vs i32 type comparison.	2022-04-22 08:09:58 +05:30
Utils.h	[MLIR][TORCH] Add E2E support for max_pool2d_with_indices op	2022-04-18 21:05:19 +05:30