Merge pull request #125 from uTensor/fix-tensor-name

Fix tensor name
uTensor · Oct 28, 2020 · a54fa40 · a54fa40
2 parents 671f2c6 + 6b94245
commit a54fa40
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/utensor_cgen/frontend/tflite.py b/utensor_cgen/frontend/tflite.py
@@ -80,6 +80,12 @@ def _build_tensor_map(self, fb_model, ugraph):
     for idx in range(0, subgraph.TensorsLength()):
       tensor = subgraph.Tensors(idx)
       tensor_name = tensor.Name().decode('utf8')
+      illegal_chars = ';'
+      if any(c in tensor_name for c in illegal_chars):
+        logger.warning(
+          f'Unexpected character founded in tensor name {tensor_name}, will be replaced or pruned'
+        )
+      tensor_name = tensor_name.replace(';', '__')
       if tensor_name is "" or None:
         tensor_name = "tensor_" + str(idx)