llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
1,570 Commits
47 Branches
1195 Tags
60 MiB
Commit Graph

6 Commits (68f568b7041dd216a0d34901fcee7375e57840c6)

Author SHA1 Message Date
AmosLewis 22a5067242 [TOSA] Add more tosa::cast type support 2022-11-16 09:53:10 +05:30
Vivek Khandelwal d3cc3f1aff [tosa] Add lowering for aten.to.dtype and aten._to_copy op 
This commit adds the TorchToTosa lowering for `aten.to.dtype` and
`aten._to_copy` op.

Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
 2022-10-06 12:00:25 +05:30
Suraj Sudhir eb06d21765
[tosa] Implement conv2d support (#541) 
Signed-off-by: Suraj Sudhir <suraj.sudhir@arm.com>
 2022-01-26 19:16:13 -08:00
Anup Gangwar f8080bd1c5
* [tosa] Support for AtenRsubScalarOp for scalar constants (#531) 
* [tosa] Support for AtenCeilOp and AtenReciprocalOp
* [tosa] Support for comparator ops, Aten[Gt|Lt|Eq][Tensor|Scalar]Op with scalar constant
* [tosa] Support for Scalar variants of Aten[Mul|Div|Add|Sub] Ops with scalar constants

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

Co-authored-by: Anup Gangwar <anup.gangwar@arm.com>
 2022-01-20 10:58:30 -08:00
Suraj Sudhir edf4a0e729
[tosa] Add more common utility functions (#525) 
- Common code as TF repository, being moved to MLIR core.
- Will support further legalizations to be published.

Signed-off-by: Suraj Sudhir <suraj.sudhir@arm.com>
 2022-01-14 13:57:27 -08:00
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