Skip to content

ONNX dev

Cyril Moineau requested to merge DevONNX into master

Multiple devs:

ONNX Export:

  • Add support for PaddedConv, PaddedConvDW, PaddedMaxPool
  • Add mecanism to support for different opset
  • Add inputs_dims and output_dims parameters to avoid having to do a forward_dim
  • Fix bug when graphview is not connected to a producer as an input
  • Update ONNX export to explore the graph in a tological order

ONNX Import:

  • Add mecanism to support for different opset
  • Fix padding indexation error when imported
  • Fix naming error when importing constant

Misc:

  • Add a function aidge_onnx.check_onnx_validity(onnx_path) to check generated ONNX validity.
  • Add a unittest to import a MobileNetv1 export it and check the validity of the export.
  • Update the package with operator refactoring in aidge_core
Edited by Cyril Moineau

Merge request reports

Loading