63 lines
1.8 KiB
Bash
63 lines
1.8 KiB
Bash
#!/usr/bin/env bash
|
|
# Copyright 2016 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.
|
|
# ==============================================================================
|
|
|
|
set -e
|
|
|
|
go get github.com/golang/protobuf/proto
|
|
go get github.com/golang/protobuf/protoc-gen-go
|
|
|
|
if [ -z "${GOPATH}" ]
|
|
then
|
|
GOPATH=$(go env GOPATH)
|
|
fi
|
|
|
|
cd $(dirname $0)
|
|
for g in $(echo "${GOPATH//:/ }"); do
|
|
TF_DIR="${g}/src/github.com/tensorflow/tensorflow"
|
|
PROTOC="${TF_DIR}/bazel-out/host/bin/external/protobuf/protoc"
|
|
if [ -x "${PROTOC}" ]; then
|
|
break
|
|
fi
|
|
done
|
|
|
|
if [ ! -x "${PROTOC}" ]
|
|
then
|
|
set +e
|
|
PATH_PROTOC=$(which protoc)
|
|
if [ ! -x "${PATH_PROTOC}" ]
|
|
then
|
|
echo "Protocol buffer compiler protoc not found in PATH or in ${PROTOC}"
|
|
echo "Perhaps build it using:"
|
|
echo "bazel build --config opt @com_google_protobuf//:protoc"
|
|
exit 1
|
|
fi
|
|
PROTOC=$PATH_PROTOC
|
|
set -e
|
|
fi
|
|
|
|
# Ensure that protoc-gen-go is available in $PATH
|
|
# Since ${PROTOC} will require it.
|
|
export PATH=$PATH:${GOPATH}/bin
|
|
mkdir -p ../vendor
|
|
for FILE in ${TF_DIR}/tensorflow/core/framework/*.proto \
|
|
${TF_DIR}/tensorflow/core/protobuf/*.proto \
|
|
${TF_DIR}/tensorflow/stream_executor/*.proto; do
|
|
${PROTOC} \
|
|
-I ${TF_DIR} \
|
|
--go_out=../vendor \
|
|
$FILE
|
|
done
|