Change ReduceMean attribute 'axes' from static array to dynamic vector
Change axes attribute type to allow runtime deduction of 'axes' dimensions in the ReduceMean used in MSE loss function
parent
e2a851e5
No related branches found
No related tags found
Showing
- include/aidge/backend/cpu/operator/ReduceMeanImpl.hpp 81 additions, 57 deletionsinclude/aidge/backend/cpu/operator/ReduceMeanImpl.hpp
- include/aidge/backend/cpu/operator/ReduceMeanImpl_forward_kernels.hpp 43 additions, 33 deletions...e/backend/cpu/operator/ReduceMeanImpl_forward_kernels.hpp
- src/operator/ReduceMeanImpl.cpp 65 additions, 49 deletionssrc/operator/ReduceMeanImpl.cpp
Loading
Please register or sign in to comment