aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/go/genop/generate.sh
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/go/genop/generate.sh')
-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} \