MLBEDSW-4840 Move setting of input indices to tflite reader
Mapping to internal input indexing has been added to
tflite_reader.py and tosa_reader.py.
And the other way around in tflite_writer.py.
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I4d8596e747cfa7c4203884c4e785eb1977e2bcc1
diff --git a/ethosu/vela/tosa_reader.py b/ethosu/vela/tosa_reader.py
index ac0b396..e51ead1 100644
--- a/ethosu/vela/tosa_reader.py
+++ b/ethosu/vela/tosa_reader.py
@@ -25,6 +25,7 @@
from .nn_graph import Subgraph
from .operation import Op
from .operation import Operation
+from .reader_util import align_tensor_indices_to_nng
from .reader_util import clone_and_reshape_tensor
from .reader_util import decode_str
from .reader_util import fixup_tensors
@@ -104,8 +105,8 @@
name = "unknown_op_name"
if len(outputs):
name = outputs[0].name
+ inputs = align_tensor_indices_to_nng(op_type, indices, inputs)
op = Operation(op_type, name)
- op.type.info.indices = indices
op.op_index = op_index
op.inputs = inputs
op.outputs = outputs