Skip to content
Commit 0cbb166c authored by Erik Andersson's avatar Erik Andersson
Browse files

MLBEDSW-3571: Sum and FC should not crash when asking for keep_dims.



Previously the keep_dims or keep_num_dims attribute was not supported for Sum and Fully Connected operators and would thus crash for certain tests. With this update, the attribute is extracted correctly and saved to the optimised tflite file.

Signed-off-by: Erik Andersson's avatarerik.andersson@arm.com <erik.andersson@arm.com>
Change-Id: If33487f6d299bb99788bb3d13332b842ba961641
parent 460c6896
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment