diff options
author | Dwight Lidman <dwight.lidman@arm.com> | 2020-08-17 11:56:10 +0200 |
---|---|---|
committer | tim.hall <tim.hall@arm.com> | 2020-08-19 11:28:35 +0000 |
commit | 2573311dbed6728fddba8bcb0cd03c35c45bbc03 (patch) | |
tree | 25095cd37a90bb81a7f5af1583a62fc2642f6619 /ethosu/vela/tflite_mapping.py | |
parent | fa34c6f4eebec25814c1e620a85721416f4d4ce3 (diff) | |
download | ethos-u-vela-2573311dbed6728fddba8bcb0cd03c35c45bbc03.tar.gz |
MLBEDSW-2729: Add restrictions for shapeless tensors
Vela often fails when encountering operators that have
inputs or outputs with shape == []. Only for elementwise
ops where shape is broadcasted from IFM2 to IFM1 is this
supported.
This commit adds a restriction which places ops with
shape [] tensors on the CPU except in the special case
of broadcasting for elemwise ops.
Signed-off-by: Dwight Lidman <dwight.lidman@arm.com>
Change-Id: I5b0855233e3b83870209f4da00fb2dbd0184fee0
Diffstat (limited to 'ethosu/vela/tflite_mapping.py')
0 files changed, 0 insertions, 0 deletions