diff options
Diffstat (limited to 'tools/tflitefile_tool/tflite/Tensor.py')
-rwxr-xr-x[-rw-r--r--] | tools/tflitefile_tool/tflite/Tensor.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tools/tflitefile_tool/tflite/Tensor.py b/tools/tflitefile_tool/tflite/Tensor.py index 120cce50b..e5f13301c 100644..100755 --- a/tools/tflitefile_tool/tflite/Tensor.py +++ b/tools/tflitefile_tool/tflite/Tensor.py @@ -97,8 +97,8 @@ def TensorStartShapeVector(builder, numElems): return builder.StartVector(4, numElems, 4) -def TensorAddType(builder, typeSlot): - builder.PrependInt8Slot(1, typeSlot, 0) +def TensorAddType(builder, type): + builder.PrependInt8Slot(1, type, 0) def TensorAddBuffer(builder, buffer): |