Add transpose function for 4D tensors and related templates
Context
New transpose OP for export_cpp
Modified files
-
transpose.hpp
new kernel transpose; -
operator.py
, to take in case an export of node Transpose; -
transpose_ND_forward.jinja
and 'transpose_ND_config.jinja' to generate the code.
TODO
-
Unit-Tests : waiting for the feature to compare results between the export, onnx_runtime and aidge from @pineapple
Edited by Cyril Moineau
Merge request reports
Activity
Filter activity
Please register or sign in to reply