projects / platform / core / ml / nntrainer.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 5e11ba1) | patch
[layer] added start/end dimension in flatten layer
authorhyeonseok <hs89.lee@samsung.com>
Thu, 30 May 2024 11:32:28 +0000 (20:32 +0900)
committerMyungJoo Ham <myungjoo.ham@samsung.com>
Wed, 3 Jul 2024 06:19:08 +0000 (15:19 +0900)
commit6f98e73dad5477942c2d9888a594127899ef4027
treee003d3fd4679a209c2c2fd39fbe5dfa52b88d640tree | snapshot
parent5e11ba13abdf651343baa9b6c2a76a83db02e921commit | diff
[layer] added start/end dimension in flatten layer

 - For now flatten layer flatten all dimension except batch.
   This commit will be able to flatten only the sub dimensions

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