Bladeren bron

Build flatc compiler locally

Lutz Roeder 8 jaren geleden
bovenliggende
commit
b1a7b39dba
2 gewijzigde bestanden met toevoegingen van 29 en 4 verwijderingen
  1. 1 4
      tools/tf-generate
  2. 28 0
      tools/tflite-generate

+ 1 - 4
tools/tf-generate

@@ -1,3 +1,4 @@
+
 #!/bin/bash
 
 mkdir -p ../third_party
@@ -34,9 +35,5 @@ echo "Generate '../src/tf.js'"
     ../third_party/tensorflow/tensorflow/core/framework/tensor.proto \
     ../third_party/tensorflow/tensorflow/core/framework/resource_handle.proto \
 
-echo "Generate '../src/tflite.js'"
-flatc --js ../third_party/tensorflow/tensorflow/contrib/lite/schema/schema.fbs 
-mv ./schema_generated.js ../src/tflite.js
-
 echo "Generate '../src/tf-operator.json'"
 python tf-operator-json.py

+ 28 - 0
tools/tflite-generate

@@ -0,0 +1,28 @@
+#!/bin/bash
+
+mkdir -p ../third_party
+
[email protected]:google/flatbuffers.git
+
+if [ -d "../third_party/flatbuffers" ]; then
+    pushd "../third_party/flatbuffers" > /dev/null
+    echo "Fetch ${repository}..."
+    git fetch -p
+    echo "Reset ${repository}..."
+    git reset --hard origin/master
+    popd > /dev/null
+else
+    pushd "../third_party" > /dev/null
+    echo "Clone ${repository}..."
+    git clone --recursive ${repository}
+    popd > /dev/null
+fi
+
+pushd "../third_party/flatbuffers" > /dev/null
+cmake -G "Unix Makefiles"
+make
+popd > /dev/null
+
+echo "Generate '../src/tflite.js'"
+../third_party/flatbuffers/flatc --js ../third_party/tensorflow/tensorflow/contrib/lite/schema/schema.fbs 
+mv ./schema_generated.js ../src/tflite.js