torch-mlir

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