torch-mlir/lib/Dialect/Torch/Transforms
Gaurav Shukla 4b911ada40 [LINALG] Add E2E support for `aten.mean.dim` op
- This commit adds support for `aten.mean.dim` op.
- It also adds a new test script `stats.py` for statistics related ops.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
2022-05-04 20:11:42 +05:30
..
AdjustCallingConventions.cpp llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
CMakeLists.txt Introduce new shape library design. 2022-03-15 12:41:58 -07:00
DecomposeComplexOps.cpp [LINALG] Add E2E support for `aten.mean.dim` op 2022-05-04 20:11:42 +05:30
DropShapeCalculations.cpp llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
GlobalizeObjectGraph.cpp llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
InlineGlobalSlots.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
MaximizeValueSemantics.cpp [MLIR][TORCH] Add E2E support for aten.to.dtype_layout op 2022-05-03 12:48:58 +05:30
PassDetail.h llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
Passes.cpp llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
PrepareForGlobalizeObjectGraph.cpp llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
ReduceOpVariants.cpp llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
RefinePublicReturn.cpp llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
RefineTypes.cpp [MLIR][TORCH] Add E2E support for aten.to.dtype_layout op 2022-05-03 12:48:58 +05:30
ReifyShapeCalculations.cpp llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
ShapeLibrary.cpp [MLIR][TORCH] Add E2E support for aten.to.dtype_layout op 2022-05-03 12:48:58 +05:30
SimplifyShapeCalculations.cpp llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00