torch-mlir/lib/Conversion/TorchToTosa
Anup Gangwar 756b75fb2d
[tosa] Support for some ops and fix for Issue #532 (#575)
* [tosa] Support for AtenNe[Tensor|Scalar]Op, AtenLog2Op,
AtenBitwiseAndTensorOp, AtenSquareOp and AtenThresholdOp
* Fix for Issue #532 - Mixed input types for few ops and updated few
tests to use i32 instead of i64

Signed-off-by: Anup Gangwar <anup.gangwar@arm.com>

Co-authored-by: Anup Gangwar <anup.gangwar@arm.com>
2022-02-11 12:30:02 -08:00
..
CMakeLists.txt [tosa] Add Torch reduction operators 2021-12-03 09:01:48 -08:00
TorchToTosa.cpp [tosa] Support for some ops and fix for Issue #532 (#575) 2022-02-11 12:30:02 -08:00
TosaLegalizeCommon.cpp [tosa] Add Torch reduction operators 2021-12-03 09:01:48 -08:00
TosaLegalizeUtils.cpp [tosa] Implement conv2d support (#541) 2022-01-26 19:16:13 -08:00