projects / platform / upstream / armnn.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: c01b391) | patch
IVGCVSW-2996 Add Reshape layer to ParseFullyConnected in TfLite parser
authorNarumol Prangnawarat <narumol.prangnawarat@arm.com>
Wed, 24 Apr 2019 14:52:20 +0000 (15:52 +0100)
committerMatteo Martincigh <matteo.martincigh@arm.com>
Mon, 29 Apr 2019 08:36:55 +0000 (08:36 +0000)
commit501f4d4efff78f890602d062709126f9a294a352
tree259dcc6bb6fcd895841c276ebc5025c756f656e3tree | snapshot
parentc01b39146d4b67159ef95a2ac2f88f6536320890commit | diff
IVGCVSW-2996 Add Reshape layer to ParseFullyConnected in TfLite parser
when input is > 2D to flatten the input to 2D [batch_size, input_size]

Change-Id: Id9d9ff996225c7d0938204ae0ceb330a11e264f5
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
src/armnnTfLiteParser/TfLiteParser.cpp diff | blob | history
src/armnnTfLiteParser/test/FullyConnected.cpp diff | blob | history
Domain: Machine Learning / ML Framework;