[MatMul] Add implementation of MatMul export as MatMulCPP
Compare changes
Files
4
aidge_export_cpp/kernels/matmul.hpp
0 → 100644
+ 33
− 0
Add implementation of export for MAtMul in exportCPP.
Choice to provide a dedicated matmul.hpp kernel as mapping to FC does not really scale to 2-Dims and N-dims matmults,
Notes:
np.matmul
which does not support scalar inputs, don't know if it was expected.
This seems reasonable to me, but for instance the backend_cpu does not support it.Add in operator export file operators.py
the single class MatMulCPP
,
plus the templates/kernel files.
No major modification. Feature added as Matmul is now supported in export CPP.
No TODO forecasted, as explained above I prefer to add the >2D inputs support at a later stage if you don't mind.
Copyright © Eclipse Foundation, Inc. All Rights Reserved. Privacy Policy | Terms of Use | Copyright Agent