torch-mlir/lib/Conversion/TorchToTosa
Suraj Sudhir c9c9b68d1f [tosa] Add Torch reduction operators
- Supports variants with multiple dims, one dim, all dime
- Leverages legalize_common and legalize_utils code from
TensorFlow-TOSA work

Signed-off-by: Suraj Sudhir <suraj.sudhir@arm.com>
2021-12-03 09:01:48 -08:00
..
CMakeLists.txt [tosa] Add Torch reduction operators 2021-12-03 09:01:48 -08:00
TorchToTosa.cpp [tosa] Add Torch reduction operators 2021-12-03 09:01:48 -08:00
TosaLegalizeCommon.cpp [tosa] Add Torch reduction operators 2021-12-03 09:01:48 -08:00
TosaLegalizeUtils.cpp [tosa] Add Torch reduction operators 2021-12-03 09:01:48 -08:00