-
- Downloads
move every node with a backward in gradient accumulation mode
parent
6ca3699c
No related branches found
No related tags found
Showing
- include/aidge/backend/cuda/operator/ReduceImpl_CUDA_kernels.hpp 3 additions, 1 deletion...e/aidge/backend/cuda/operator/ReduceImpl_CUDA_kernels.hpp
- src/operator/AddImpl.cpp 3 additions, 1 deletionsrc/operator/AddImpl.cpp
- src/operator/AvgPoolingImpl.cpp 4 additions, 2 deletionssrc/operator/AvgPoolingImpl.cpp
- src/operator/BatchNormImpl.cpp 2 additions, 2 deletionssrc/operator/BatchNormImpl.cpp
- src/operator/ConvImpl.cpp 1 addition, 1 deletionsrc/operator/ConvImpl.cpp
- src/operator/DivImpl.cpp 1 addition, 1 deletionsrc/operator/DivImpl.cpp
- src/operator/FCImpl.cpp 6 additions, 27 deletionssrc/operator/FCImpl.cpp
- src/operator/GlobalAveragePoolingImpl.cpp 1 addition, 1 deletionsrc/operator/GlobalAveragePoolingImpl.cpp
- src/operator/LnImpl.cpp 1 addition, 2 deletionssrc/operator/LnImpl.cpp
- src/operator/MaxPoolingImpl.cpp 1 addition, 1 deletionsrc/operator/MaxPoolingImpl.cpp
- src/operator/MulImpl.cpp 2 additions, 2 deletionssrc/operator/MulImpl.cpp
- src/operator/PadImpl.cpp 1 addition, 1 deletionsrc/operator/PadImpl.cpp
- src/operator/PadImpl_CUDA_kernels.cu 2 additions, 1 deletionsrc/operator/PadImpl_CUDA_kernels.cu
- src/operator/ReLUImpl.cpp 5 additions, 2 deletionssrc/operator/ReLUImpl.cpp
- src/operator/ReduceImpl_CUDA_kernels.cu 47 additions, 8 deletionssrc/operator/ReduceImpl_CUDA_kernels.cu
- src/operator/ReduceMeanImpl.cpp 9 additions, 1 deletionsrc/operator/ReduceMeanImpl.cpp
- src/operator/ReduceSumImpl.cpp 8 additions, 2 deletionssrc/operator/ReduceSumImpl.cpp
- src/operator/ReshapeImpl.cpp 2 additions, 1 deletionsrc/operator/ReshapeImpl.cpp
- src/operator/ScalingImpl_CUDA_kernels.cu 4 additions, 2 deletionssrc/operator/ScalingImpl_CUDA_kernels.cu
- src/operator/SigmoidImpl.cpp 4 additions, 1 deletionsrc/operator/SigmoidImpl.cpp
Loading
Please register or sign in to comment