Skip to content

Add transpose function for 4D tensors and related templates

Wissam Boussella requested to merge wboussella/aidge_export_cpp:transpose_op into dev

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

Loading