mirror of https://github.com/llvm/torch-mlir
14ef05a292
- Add Torch to TOSA legalization for the following reduction ops: + aten.min.dim + aten.min + aten.max + aten.prod + aten.prod.dim_int + aten.all.dim - Add dtype casting support for reduce sum and prod ops - Extend aten.max.dim legalization to a template to support aten.min.dim legalization - Update end-to-end tests sets in xfail_sets.py Signed-off-by: Justin Ngo <justin.ngo@arm.com> Change-Id: I854dd6c0c55e570c1fb7242f20c85cf64d6e7fe0 Signed-off-by: Justin Ngo <justin.ngo@arm.com> |
||
---|---|---|
.. | ||
CMakeLists.txt | ||
TorchToTosa.cpp | ||
TosaLegalizeCommon.cpp | ||
TosaLegalizeUtils.cpp |