Clean up Makefile based build scripts for TFLite iOS
For TFLite iOS, bazel has been the recommended way to build the static framework and the Makefile based script is not even meeting feature parity with the bazel build (e.g., no GPU delegate support). PiperOrigin-RevId: 317587116 Change-Id: Idbdf60687f35e1ef0bdb074668317818eeaa95e5
This commit is contained in:
YoungSeok Yoon
TensorFlower Gardener
parent
8e9a05117b
commit
db121c7f4b
|
|
@ -1,120 +0,0 @@
|
|||
#!/bin/bash
|
||||
# Copyright 2017 The TensorFlow Authors. All Rights Reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
# ==============================================================================
|
||||
|
||||
# TODO(ycling): Refactoring - Move this script into `tools/make`.
|
||||
set -e
|
||||
|
||||
echo "Starting"
|
||||
TFLITE_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/.."
|
||||
|
||||
usage() {
|
||||
echo "Usage: $(basename "$0") [-a]"
|
||||
echo "-g build with GPU delegate"
|
||||
exit 1
|
||||
}
|
||||
|
||||
USE_GPU_DELEGATE="false"
|
||||
FRAMEWORK_NAME="tensorflow_lite"
|
||||
while getopts "g" opt_name; do
|
||||
case "$opt_name" in
|
||||
g)
|
||||
USE_GPU_DELEGATE="true"
|
||||
FRAMEWORK_NAME="tensorflow_lite_gpu"
|
||||
;;
|
||||
*) usage;;
|
||||
esac
|
||||
done
|
||||
shift $((OPTIND - 1))
|
||||
readonly USE_GPU_DELEGATE
|
||||
readonly FRAMEWORK_NAME
|
||||
|
||||
if [ $USE_GPU_DELEGATE == "true" ] ; then
|
||||
for filename in metal_delegate.h libmetal_delegate.a ; do
|
||||
if [[ ! -f "${TFLITE_DIR}/delegates/gpu/${filename}" ]] ; then
|
||||
echo "File ${TFLITE_DIR}/delegates/gpu/${filename} doesn't exist."
|
||||
echo "It's required for building TFLite Framework with GPU. Aborting."
|
||||
exit 1
|
||||
fi
|
||||
done
|
||||
fi
|
||||
|
||||
TMP_DIR=$(mktemp -d)
|
||||
echo "Package dir: " $TMP_DIR
|
||||
FW_DIR=$TMP_DIR/tensorflow_lite_ios_frameworks
|
||||
FW_DIR_TFLITE=$FW_DIR/$FRAMEWORK_NAME.framework
|
||||
FW_DIR_TFLITE_HDRS=$FW_DIR_TFLITE/Headers
|
||||
|
||||
echo "Creating target Headers directories"
|
||||
mkdir -p $FW_DIR_TFLITE_HDRS
|
||||
|
||||
echo "Headers, populating: TensorFlow Lite"
|
||||
cd $TFLITE_DIR/../..
|
||||
|
||||
find tensorflow/lite -name '*.h' \
|
||||
-not -path 'tensorflow/lite/tools/*' \
|
||||
-not -path 'tensorflow/lite/examples/*' \
|
||||
-not -path 'tensorflow/lite/gen/*' \
|
||||
-not -path 'tensorflow/lite/toco/*' \
|
||||
-not -path 'tensorflow/lite/nnapi/*' \
|
||||
-not -path 'tensorflow/lite/java/*' \
|
||||
| tar -cf $FW_DIR_TFLITE_HDRS/tmp.tar -T -
|
||||
cd $FW_DIR_TFLITE_HDRS
|
||||
tar xf tmp.tar
|
||||
rm -f tmp.tar
|
||||
|
||||
echo "Headers, populating: Flatbuffer"
|
||||
cd $TFLITE_DIR/tools/make/downloads/flatbuffers/include/
|
||||
find . -name '*.h' | tar -cf $FW_DIR_TFLITE_HDRS/tmp.tar -T -
|
||||
cd $FW_DIR_TFLITE_HDRS
|
||||
tar xf tmp.tar
|
||||
rm -f tmp.tar
|
||||
|
||||
cd $TFLITE_DIR/../..
|
||||
echo "Generate LICENSE files and copy to target"
|
||||
bazel build //tensorflow/tools/lib_package:clicenses_generate
|
||||
cp $TFLITE_DIR/../../LICENSE $FW_DIR_TFLITE
|
||||
cp $TFLITE_DIR/../../bazel-bin/tensorflow/tools/lib_package/THIRD_PARTY_TF_C_LICENSES \
|
||||
$FW_DIR_TFLITE
|
||||
|
||||
echo "Copying static libraries"
|
||||
# Note: There must be a static library with the same name
|
||||
# as the framework name.
|
||||
cp $TFLITE_DIR/tools/make/gen/lib/libtensorflow-lite.a \
|
||||
$FW_DIR_TFLITE/$FRAMEWORK_NAME
|
||||
if [ $USE_GPU_DELEGATE == "true" ] ; then
|
||||
cp "${TFLITE_DIR}/delegates/gpu/libmetal_delegate.a" \
|
||||
$FW_DIR_TFLITE/libmetal_delegate.a
|
||||
fi
|
||||
|
||||
# This is required, otherwise they interfere with the documentation of the
|
||||
# pod at cocoapods.org.
|
||||
echo "Remove all README files"
|
||||
cd $FW_DIR_TFLITE_HDRS
|
||||
find . -type f -name README\* -exec rm -f {} \;
|
||||
find . -type f -name readme\* -exec rm -f {} \;
|
||||
|
||||
TARGET_GEN_LOCATION="$TFLITE_DIR/gen/ios_frameworks"
|
||||
echo "Moving results to target: " $TARGET_GEN_LOCATION
|
||||
cd $FW_DIR
|
||||
zip -q -r $FRAMEWORK_NAME.framework.zip $FRAMEWORK_NAME.framework -x .DS_Store
|
||||
rm -rf $TARGET_GEN_LOCATION
|
||||
mkdir -p $TARGET_GEN_LOCATION
|
||||
cp -r $FRAMEWORK_NAME.framework.zip $TARGET_GEN_LOCATION
|
||||
|
||||
echo "Cleaning up"
|
||||
rm -rf $TMP_DIR
|
||||
|
||||
echo "Finished"
|
||||
|
|
@ -1,4 +1,4 @@
|
|||
#!/bin/bash -x
|
||||
#!/bin/bash +x
|
||||
# Copyright 2017 The TensorFlow Authors. All Rights Reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
|
|
@ -14,7 +14,15 @@
|
|||
# limitations under the License.
|
||||
# ==============================================================================
|
||||
|
||||
set -e
|
||||
echo "========================================================================="
|
||||
echo "WARNING: This build script is deprecated and no longer maintained. Please"
|
||||
echo " refer to the iOS build guide to learn how to build the latest "
|
||||
echo " version of TFLite static framework for iOS using bazel. "
|
||||
echo " https://www.tensorflow.org/lite/guide/build_ios "
|
||||
echo "========================================================================="
|
||||
sleep 5s
|
||||
|
||||
set -ex
|
||||
|
||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
cd "$SCRIPT_DIR/../../../.."
|
||||
|
|
|
|||
Loading…
Reference in New Issue