Adding MatMul operator
All threads resolved!
All threads resolved!
Compare changes
Files
4+ 90
− 10
@@ -6,9 +6,8 @@
@@ -19,15 +18,96 @@ void matmul_forward (
\ No newline at end of file
While trying to export the ConvNeXt onnx model to cpp, we encountered several issues. we needed to add broadcasting support for the matmul operator.
matmul.hpp
add broadcasting for matmul;
matmul_config.jinja
, and matmul_forward.jinja
to include for matmul operator;
operator.py
, register matmul operator;
test_export.cpp
, add unit tests;
Add matmul broadcasting and unit tests.
Added matmul broadcasting for exporting ConvNeXt model with aidge_export_cpp;
Copyright © Eclipse Foundation, Inc. All Rights Reserved. Privacy Policy | Terms of Use | Copyright Agent