projects / platform / core / ml / nntrainer.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 27907c8) | patch
[layer] Add support for reduce mean layer
authorParichay Kapoor <pk.kapoor@samsung.com>
Thu, 25 Nov 2021 06:31:08 +0000 (15:31 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Thu, 2 Dec 2021 06:54:18 +0000 (15:54 +0900)
commit899cc267ac3b635ff158128d34c429a3a3c33cef
tree5c908dbe827f36908b96f86987f2763f17a33fb9tree | snapshot
parent27907c81bb6ae1936537c24b2e31a9e2b0bfccf9commit | diff
[layer] Add support for reduce mean layer

This patch adds support for reduce_mean layer with forwarding and
backwarding implementation.
Basic unittests are added.

Golden unittests will come in the next patch.

Signed-off-by: Parichay Kapoor <pk.kapoor@samsung.com>
nntrainer/layers/meson.build diff | blob | history
nntrainer/layers/reduce_mean_layer.cpp [new file with mode: 0644] blob
nntrainer/layers/reduce_mean_layer.h [new file with mode: 0644] blob
test/unittest/layers/unittest_layers_reduce_mean.cpp [new file with mode: 0644] blob
test/unittest/layers/unittest_layers_reshape.cpp diff | blob | history
Domain: Machine Learning / ML Framework;