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/Conversion/TorchToTosa
History
Suraj Sudhir 0188ca5498
[tosa] Implement matmul, mm and bmm support (#526) 
- Also handles braodcasting n-D tensors, dynamic shapes

Signed-off-by: Suraj Sudhir <suraj.sudhir@arm.com>
 		2022-01-18 13:37:32 -08:00
..
CMakeLists.txt [tosa] Add Torch reduction operators 2021-12-03 09:01:48 -08:00
TorchToTosa.cpp [tosa] Implement matmul, mm and bmm support (#526) 2022-01-18 13:37:32 -08:00
TosaLegalizeCommon.cpp [tosa] Add Torch reduction operators 2021-12-03 09:01:48 -08:00
TosaLegalizeUtils.cpp [tosa] Add more common utility functions (#525) 2022-01-14 13:57:27 -08:00