[Model] Fix tensor size calculation
authorDongju Chae <dongju.chae@samsung.com>
Wed, 4 Aug 2021 07:13:58 +0000 (16:13 +0900)
committer채동주/On-Device Lab(SR)/Staff Engineer/삼성전자 <dongju.chae@samsung.com>
Thu, 5 Aug 2021 00:33:27 +0000 (09:33 +0900)
commit06db8ad8763b415bd0a68209b137da250921b0c3
tree43829c9ab15977208df6ba7d126632e0cc992dbf
parentcb489781d9e97b9f65ec2cf8a87743fec7dda1d6
[Model] Fix tensor size calculation

This patch fixes tensor size calculation.
The special handling is applied NONE/TRIV2 layout only.

Signed-off-by: Dongju Chae <dongju.chae@samsung.com>
src/core/ne-handler.cc
src/core/ne-model.cc