[Transform] refactor transformation interface and fix tests
Showing
- src/finn/core/modelwrapper.py 6 additions, 15 deletionssrc/finn/core/modelwrapper.py
- src/finn/transformation/__init__.py 17 additions, 5 deletionssrc/finn/transformation/__init__.py
- src/finn/transformation/batchnorm_to_affine.py 67 additions, 64 deletionssrc/finn/transformation/batchnorm_to_affine.py
- src/finn/transformation/fold_constants.py 29 additions, 26 deletionssrc/finn/transformation/fold_constants.py
- src/finn/transformation/general.py 56 additions, 47 deletionssrc/finn/transformation/general.py
- src/finn/transformation/infer_datatypes.py 10 additions, 7 deletionssrc/finn/transformation/infer_datatypes.py
- src/finn/transformation/infer_shapes.py 11 additions, 8 deletionssrc/finn/transformation/infer_shapes.py
- src/finn/transformation/streamline.py 357 additions, 325 deletionssrc/finn/transformation/streamline.py
- tests/test_basic_onnx_exec.py 2 additions, 2 deletionstests/test_basic_onnx_exec.py
- tests/test_batchnorm_to_affine.py 6 additions, 6 deletionstests/test_batchnorm_to_affine.py
- tests/test_brevitas_export.py 6 additions, 6 deletionstests/test_brevitas_export.py
- tests/test_collapse_repeated_op.py 5 additions, 5 deletionstests/test_collapse_repeated_op.py
- tests/test_factor_out_mul_sign_magnitude.py 4 additions, 4 deletionstests/test_factor_out_mul_sign_magnitude.py
- tests/test_fold_constants.py 6 additions, 6 deletionstests/test_fold_constants.py
- tests/test_general_transformation.py 2 additions, 2 deletionstests/test_general_transformation.py
- tests/test_infer_datatypes.py 9 additions, 9 deletionstests/test_infer_datatypes.py
- tests/test_infer_shapes.py 2 additions, 2 deletionstests/test_infer_shapes.py
- tests/test_is_linear.py 3 additions, 3 deletionstests/test_is_linear.py
- tests/test_mixed_onnx_exec.py 2 additions, 2 deletionstests/test_mixed_onnx_exec.py
- tests/test_move_add_past_mul.py 6 additions, 6 deletionstests/test_move_add_past_mul.py
Loading
Please register or sign in to comment