projects / platform / core / ml / nnfw.git / commitdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | patch | inline | side by side (parent: 7050f92)
[tflkit] Fix bug tflite_convert.sh for version<2 (#3704)
author오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Tue, 4 Dec 2018 10:40:07 +0000 (19:40 +0900)
committer박세희/동작제어Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Tue, 4 Dec 2018 10:40:07 +0000 (19:40 +0900)
Fix bug tflite_convert.sh for version<2 setting allow_custom_ops

Signed-off-by: Hyeongseok Oh <hseok82.oh@samsung.com>
tools/tflkit/tflite_convert.sh patch | blob | history

diff --git a/tools/tflkit/tflite_convert.sh b/tools/tflkit/tflite_convert.sh
index 1694c3c..2056797 100755 (executable)
--- a/tools/tflkit/tflite_convert.sh
+++ b/tools/tflkit/tflite_convert.sh
@@ -125,7 +125,7 @@ CUR_DIR=$(pwd)
       --input_arrays="$INPUT" \
       --input_shapes="${INPUT_SHAPE_LIST[$i]}" \
       --output_arrays="$OUTPUT" \
-      --allow_custom_ops=true
+      --allow_custom_ops
     fi
   done
   popd
Domain: Machine Learning / ML Framework;