experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Internal change.

Browse Source 
PiperOrigin-RevId: 196024130
This commit is contained in:
Anna R 2018-05-09 13:55:47 -07:00 committed by TensorFlower Gardener
parent ec0ef29835
commit 5d47c53adb
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tensorflow/tools/pip_package/build_pip_package.sh
View File

@ -53,6 +53,7 @@ function main() {
PKG_NAME_FLAG=""
GPU_BUILD=0
NIGHTLY_BUILD=0
PROJECT_NAME=""
while true; do
if [[ "$1" == "--nightly_flag" ]]; then
NIGHTLY_BUILD=1
@ -60,6 +61,12 @@ function main() {
GPU_BUILD=1
elif [[ "$1" == "--gpudirect" ]]; then
PKG_NAME_FLAG="--project_name tensorflow_gpudirect"
elif [[ "$1" == "--project_name" ]]; then
shift
if [[ -z "$1" ]]; then
break
fi
PROJECT_NAME="$1"
fi
shift
@ -68,7 +75,9 @@ function main() {
fi
done
if [[ ${NIGHTLY_BUILD} == "1" && ${GPU_BUILD} == "1" ]]; then
if [[ -n ${PROJECT_NAME} ]]; then
PKG_NAME_FLAG="--project_name ${PROJECT_NAME}"
elif [[ ${NIGHTLY_BUILD} == "1" && ${GPU_BUILD} == "1" ]]; then
PKG_NAME_FLAG="--project_name tf_nightly_gpu"
elif [[ ${NIGHTLY_BUILD} == "1" ]]; then
PKG_NAME_FLAG="--project_name tf_nightly"