projects / platform / core / ml / nntrainer.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: e410114) | patch
[split_layer] enhance split layer to split input dimension by given number
authorhyeonseok lee <hs89.lee@samsung.com>
Wed, 26 Oct 2022 06:08:48 +0000 (15:08 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Fri, 18 Nov 2022 01:00:19 +0000 (10:00 +0900)
commit2e246c84ca2f608ae0b10bc21ff271caad471248
tree08d74829c81dc0029af7324c6a280de027f6e91ctree | snapshot
parente41011408c1dc51737e3884644c704fccee871a3commit | diff
[split_layer] enhance split layer to split input dimension by given number

 - Until now split layer split input dimension by input dimension size which makes output dimension to 1.
   But for now we can set split number which makes output dimension to (input dimension / split number).
 - In this commit given split numbers are expected to divide input dimension equally.
   Otherwise it will raise an error.

Signed-off-by: hyeonseok lee <hs89.lee@samsung.com>
nntrainer/layers/common_properties.h diff | blob | history
nntrainer/layers/split_layer.cpp diff | blob | history
nntrainer/layers/split_layer.h diff | blob | history
Domain: Machine Learning / ML Framework;