aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/go/genop
diff options
context:
space:
mode:
authorGravatar Asim Shankar <ashankar@google.com>2016-10-14 13:57:27 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-10-14 15:05:56 -0700
commit11ac5580f0f04a1b58ad08d6d6885806bfe445a1 (patch)
treec1618a609738f3f2af9ab909c0a71e8b96e73f1f /tensorflow/go/genop
parent11ac472e3f311c1d346c7d71abf721b3367645e6 (diff)
go: Ensure protoc-gen-go is available to protoc when generating op functions.
Change: 136202135
Diffstat (limited to 'tensorflow/go/genop')
-rw-r--r--tensorflow/go/genop/generate.sh3
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/go/genop/generate.sh b/tensorflow/go/genop/generate.sh
index ae49894b94..272c550fbd 100644
--- a/tensorflow/go/genop/generate.sh
+++ b/tensorflow/go/genop/generate.sh
@@ -35,6 +35,9 @@ then
PROTOC=PATH_PROTOC
fi
+# Ensure that protoc-gen-go is available in $PATH
+# Since ${PROTOC} will require it.
+export PATH=$PATH:${GOPATH}/bin
mkdir -p ./internal/proto
${PROTOC} \
-I ${TF_DIR} \