Add backward implementations
Context
This MR is dedicated to add the backward() implementation for currently supported operators on backend_cuda:
- Add
- AvgPool
- BatchNorm
- FC
- GlobalAvgPool
- MaxPool
- Pad
- Sub
- Reshape
This resolves #8 (closed)
Modified files
-
AddImpl.hpp
,AddImpl.cpp
andTest_AddImpl.cpp
add backward Impl for Add; -
AvgPoolingImpl.hpp
,AvgPoolingImpl.cpp
andTest_AvgPoolingImpl.cpp
add backward Impl for AvgPooling; -
BatchNormImpl.hpp
,BatchNormImpl.cpp
andTest_BatchNormImpl.cpp
add backward Impl for BatchNorm; -
FCImpl.hpp
,FCImpl.cpp
andTest_FCImpl.cpp
add backward Impl for FC; -
GlobalAveragePoolingImpl.hpp
,GlobalAveragePoolingImpl.cpp
andTest_GlobalAveragePoolingImpl.cpp
add backward Impl for GlobalAveragePooling; -
MaxPoolingImpl.hpp
,MaxPoolingImpl.cpp
andTest_MaxPoolingImpl.cpp
add backward Impl for MaxPooling; -
PadImpl.hpp
,PadImpl.cpp
andTest_PadImpl.cpp
add backward Impl for Pad; -
SubImpl.hpp
andSubImpl.cpp
add backward Impl for Sub; -
ReshapeImpl.hpp
,ReshapeImpl.cpp
andTest_ReshapeImpl.cpp
add backward Impl for Reshape;
TODO
-
Add backward -
AvgPool backward -
BatchNorm backward -
FC backward -
GlobalAvgPool backward -
MaxPool backward -
Pad backward -
Sub backward -
Reshape backward
Edited by Houssem ROUIS