projects / platform / core / ml / nntrainer.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 8241400) | patch
[Ahub] Fix Ahub issue
authorSeoHyungjun <hyungjun.seo@samsung.com>
Thu, 24 Aug 2023 08:10:51 +0000 (17:10 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Wed, 20 Sep 2023 06:47:40 +0000 (15:47 +0900)
commitb5adf95394d4e8785b25526856d083cb44a47996
treee09c23797614dbe968019a1a111af160ce5d7f11tree | snapshot
parent82414007fbffd780ddb72f2e744362ead609d986commit | diff
[Ahub] Fix Ahub issue

Previously, nntrainer only supported fp32. nntrainer didn't need to
change the data type, but it needs to be changed to support fp16. If
the tensor type is fp16, get input_dim of InitLayerContext through
getInputDimentions and call setDataType. For getInputDimentions,
return a const object. Added getMutableInputDimentions function
because return object of getInputDimentions cannot be modified.

Signed-off-by: SeoHyungjun <hyungjun.seo@samsung.com>
nntrainer/layers/layer_context.h diff | blob | history
test/unittest/layers/layers_golden_tests.cpp diff | blob | history
Domain: Machine Learning / ML Framework;