Vivek Khandelwal c681c3497a [MLIR][TORCH} Fix empty dim cases for the .dim ops ... This commit fixes the shape calculation for: 1.) aten.mean.dim 2.) aten.var.dim 3.) aten.sum.dim_IntList op Also, it fixes the lowering of `aten.mean.dim` and `aten.sum.dim_IntList` for handling the cases of empty dim list. Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com		2022-07-29 11:08:57 +05:30
..
CMakeLists.txt	Bump LLVM (#958)	2022-06-22 22:23:46 -05:00
DataMovement.cpp	Allow expanding and collapsing in aten::view (#1082)	2022-07-20 17:35:51 -04:00
IndirectDataMovement.cpp	[MLIR][TORCH] Add support for multiple indexing tensors for aten.index.Tensor (#1097)	2022-07-28 19:00:02 -04:00
Linear.cpp	[LINALG] Lower `aten.Matmul` to `linalg.BatchMatmul`	2022-06-25 10:58:06 +05:30
Pooling.cpp	[MLIR][TORCH] Add decomposition of aten.adaptive_avg_pool2d op	2022-05-27 07:56:37 +05:30
PopulatePatterns.h	Temporarily revert support for custom op extensions. (#944)	2022-06-14 18:24:40 -07:00
Random.cpp	Improve "pseudo" op terminology.	2022-03-15 17:57:52 -07:00
Reduction.cpp	[MLIR][TORCH} Fix empty dim cases for the .dim ops	2022-07-29 11:08:57 +05:30
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	Temporarily revert support for custom op extensions. (#944)	2022-06-14 18:24:40 -07:00
Uncategorized.cpp	Add e2e support for aten.expm1	2022-07-27 12:31:35 +05:30
Utils.cpp	[LINALG] Lower `aten.Matmul` to `linalg.BatchMatmul`	2022-06-25 10:58:06 +05:30
Utils.h	[LINALG] Lower `aten.Matmul` to `linalg.BatchMatmul`	2022-06-25 10:58:06 +05:30