Skip to content

MLBEDSW-11145: Rework of TFLite dynamic shape handling

Max Bergfelt requested to merge MLBEDSW-11145 into main

Did the following changes to correctly handle TFLite dynamic shapes:

  • Changed the TFLite semantic checks to not fail on dynamic shapes for certain operators.
  • Changed supported operators to catch dynamic shapes in non-batch dimensions and reject to CPU.
  • Added a few checks to not read empty values from the TFLite flatbuffer that can occur in some parts of the code when shapes are dynamic.

Change-Id: I59b77082619e8e64cc7d46d509ba844e8cd9e23b Signed-off-by: Max Bergfelt max.bergfelt@arm.com

Merge request reports