Add the BatchNorm train/test flag support
No related branches found
No related tags found
Checking pipeline status
Showing
- include/aidge/operator/BatchNorm.hpp 12 additions, 8 deletionsinclude/aidge/operator/BatchNorm.hpp
- python_binding/operator/pybind_BatchNorm.cpp 4 additions, 3 deletionspython_binding/operator/pybind_BatchNorm.cpp
- src/operator/BatchNorm.cpp 5 additions, 4 deletionssrc/operator/BatchNorm.cpp
- unit_tests/graph/Test_Matching.cpp 2 additions, 2 deletionsunit_tests/graph/Test_Matching.cpp
Loading
Please register or sign in to comment