[gradients] Get gradient of each layer
authorParichay Kapoor <pk.kapoor@samsung.com>
Wed, 24 Jun 2020 06:46:13 +0000 (15:46 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Thu, 25 Jun 2020 07:42:59 +0000 (16:42 +0900)
commitbb2dd7e383eda96d9502683b76c55ce72f9659f2
tree7d7b618ea036cb21942390f66cca50d12832d303
parentb06b4781c0ff0532d47ae0a25207668cd556aa17
[gradients] Get gradient of each layer

Added function to get gradient of each layer
Each layer is responsible to fill in the list containing the gradients

Signed-off-by: Parichay Kapoor <pk.kapoor@samsung.com>
nntrainer/include/layer.h
nntrainer/src/conv2d_layer.cpp
nntrainer/src/fc_layer.cpp
nntrainer/src/layer.cpp