llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Dialect/Torch
History
Vivek Khandelwal 6db513c51d
[tosa] Add support for some cases of aten.broadcast_to op (#1429) 
This commit adds support for TorchToTosa lowering of
`aten.broadcast_to` op for cases:
1.) When the rank of input and output tensor is equal.
2.) When the rank of input tensor is zero.

Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
 		2022-09-29 09:40:56 -07:00
..
IR [MLIR] Add canonicalizer for aten.slice.t op (#1413) 2022-09-26 14:35:50 -07:00
Transforms [tosa] Add support for some cases of aten.broadcast_to op (#1429) 2022-09-29 09:40:56 -07:00
Utils [tosa] Add support for some cases of aten.broadcast_to op (#1429) 2022-09-29 09:40:56 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00