Skip to content

Commit 7381df9

Browse files
committed
Fix builds for TensorFlow Lite on Mac and Windows
1 parent e1565dc commit 7381df9

1 file changed

Lines changed: 7 additions & 1 deletion

File tree

tensorflow-lite/cppbuild.sh

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,10 @@ download https://github.com/tensorflow/tensorflow/archive/v$TENSORFLOW_VERSION.t
1818
mkdir -p "$PLATFORM$EXTENSION"
1919
cd "$PLATFORM$EXTENSION"
2020
INSTALL_PATH=`pwd`
21+
TENSORFLOW_SOURCE_DIR="$INSTALL_PATH/tensorflow-$TENSORFLOW_VERSION"
22+
if which cygpath; then
23+
TENSORFLOW_SOURCE_DIR=$(cygpath -m $TENSORFLOW_SOURCE_DIR)
24+
fi
2125

2226
echo "Decompressing archives..."
2327
tar --totals -xzf ../tensorflow-$TENSORFLOW_VERSION.tar.gz || tar --totals -xzf ../tensorflow-$TENSORFLOW_VERSION.tar.gz || true
@@ -35,7 +39,8 @@ sedinplace '/#include <math.h>/a\
3539
sedinplace "s/TFLITE_VERSION_STRING/\"$TENSORFLOW_VERSION\"/g" tensorflow-$TENSORFLOW_VERSION/tensorflow/lite/core/c/c_api.cc
3640
sedinplace "s/TFLITE_EXTENSION_APIS_VERSION_STRING/\"$TENSORFLOW_VERSION\"/g" tensorflow-$TENSORFLOW_VERSION/tensorflow/lite/core/c/c_api.cc
3741
sedinplace '/private:/d' tensorflow-$TENSORFLOW_VERSION/tensorflow/lite/core/model_building.h
38-
sedinplace "s:TENSORFLOW_SOURCE_DIR \"\":TENSORFLOW_SOURCE_DIR \"$PWD/tensorflow-$TENSORFLOW_VERSION\":g" tensorflow-$TENSORFLOW_VERSION/tensorflow/lite/CMakeLists.txt
42+
sedinplace '/core\/example/d' tensorflow-$TENSORFLOW_VERSION/tensorflow/lite/CMakeLists.txt
43+
sedinplace "s:TENSORFLOW_SOURCE_DIR \"\":TENSORFLOW_SOURCE_DIR \"$TENSORFLOW_SOURCE_DIR\":g" tensorflow-$TENSORFLOW_VERSION/tensorflow/lite/CMakeLists.txt
3944

4045
if [[ ! "$PLATFORM" == windows* ]]; then
4146
mkdir -p build_flatc
@@ -123,6 +128,7 @@ sedinplace '/CMakeCCompilerId.o/d' objs
123128
sedinplace '/CMakeCXXCompilerId.o/d' objs
124129
sedinplace '/tensorflowlite_c.dir/d' objs
125130
sedinplace '/tensorflow_profiler_logger/d' objs
131+
sedinplace '/xnnpack-operator-delete.dir/d' objs
126132
# convert to DOS paths with short names to prevent exceeding MAX_PATH on Windows
127133
if which cygpath; then
128134
cygpath -d -f objs > objs.dos

0 commit comments

Comments
 (0)