Merge pull request #325 from Xilinx/feature/update-pytorch-brevitas
Update PyTorch and Brevitas
No related branches found
No related tags found
Showing
- docker/Dockerfile.finn_ci 15 additions, 4 deletionsdocker/Dockerfile.finn_ci
- docker/Dockerfile.finn_dev 19 additions, 6 deletionsdocker/Dockerfile.finn_dev
- docker/finn_entrypoint.sh 1 addition, 1 deletiondocker/finn_entrypoint.sh
- notebooks/end2end_example/cybersecurity/1-train-mlp-with-brevitas.ipynb 37 additions, 64 deletions...end_example/cybersecurity/1-train-mlp-with-brevitas.ipynb
- src/finn/builder/build_dataflow_steps.py 1 addition, 0 deletionssrc/finn/builder/build_dataflow_steps.py
- src/finn/transformation/streamline/remove.py 19 additions, 10 deletionssrc/finn/transformation/streamline/remove.py
- tests/brevitas/test_brevitas_QConv2d.py 34 additions, 6 deletionstests/brevitas/test_brevitas_QConv2d.py
- tests/brevitas/test_brevitas_avg_pool_export.py 64 additions, 65 deletionstests/brevitas/test_brevitas_avg_pool_export.py
- tests/brevitas/test_brevitas_mobilenet.py 28 additions, 0 deletionstests/brevitas/test_brevitas_mobilenet.py
- tests/brevitas/test_brevitas_non_scaled_QuantHardTanh_export.py 28 additions, 0 deletions...brevitas/test_brevitas_non_scaled_QuantHardTanh_export.py
- tests/brevitas/test_brevitas_qlinear.py 80 additions, 0 deletionstests/brevitas/test_brevitas_qlinear.py
- tests/brevitas/test_brevitas_relu_act_export.py 28 additions, 0 deletionstests/brevitas/test_brevitas_relu_act_export.py
- tests/brevitas/test_brevitas_scaled_QHardTanh_export.py 28 additions, 0 deletionstests/brevitas/test_brevitas_scaled_QHardTanh_export.py
- tests/brevitas/test_brevitas_validate_mobilenet.py 28 additions, 0 deletionstests/brevitas/test_brevitas_validate_mobilenet.py
- tests/end2end/test_end2end_bnn_pynq.py 2 additions, 6 deletionstests/end2end/test_end2end_bnn_pynq.py
- tests/end2end/test_end2end_cybsec_mlp.py 21 additions, 7 deletionstests/end2end/test_end2end_cybsec_mlp.py
- tests/transformation/streamline/test_remove_identity_ops.py 12 additions, 6 deletionstests/transformation/streamline/test_remove_identity_ops.py
Loading
Please register or sign in to comment