Allow BatchNorm implementation to handle any number of dimenions
Showing
- include/aidge/backend/cpu/operator/BatchNormImpl.hpp 1 addition, 1 deletioninclude/aidge/backend/cpu/operator/BatchNormImpl.hpp
- include/aidge/backend/cpu/operator/BatchNormImpl_kernels.hpp 3 additions, 4 deletionsinclude/aidge/backend/cpu/operator/BatchNormImpl_kernels.hpp
- src/operator/BatchNormImpl.cpp 1 addition, 3 deletionssrc/operator/BatchNormImpl.cpp
Loading
Please register or sign in to comment