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 (moved)
Modified files
-
AddImpl.hpp,AddImpl.cppandTest_AddImpl.cppadd backward Impl for Add; -
AvgPoolingImpl.hpp,AvgPoolingImpl.cppandTest_AvgPoolingImpl.cppadd backward Impl for AvgPooling; -
BatchNormImpl.hpp,BatchNormImpl.cppandTest_BatchNormImpl.cppadd backward Impl for BatchNorm; -
FCImpl.hpp,FCImpl.cppandTest_FCImpl.cppadd backward Impl for FC; -
GlobalAveragePoolingImpl.hpp,GlobalAveragePoolingImpl.cppandTest_GlobalAveragePoolingImpl.cppadd backward Impl for GlobalAveragePooling; -
MaxPoolingImpl.hpp,MaxPoolingImpl.cppandTest_MaxPoolingImpl.cppadd backward Impl for MaxPooling; -
PadImpl.hpp,PadImpl.cppandTest_PadImpl.cppadd backward Impl for Pad; -
SubImpl.hppandSubImpl.cppadd backward Impl for Sub; -
ReshapeImpl.hpp,ReshapeImpl.cppandTest_ReshapeImpl.cppadd 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