diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-12-22 12:42:59 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-22 12:46:28 -0800 |
commit | e4532d20973c4c00854492362665317551661c18 (patch) | |
tree | 398527e29bd30d39237adb4785be5069fdb646fa /tensorflow/go/genop | |
parent | 673641c2d6a27fa97ee05453d671853731a4c602 (diff) |
Merge changes from github.
PiperOrigin-RevId: 179953488
Diffstat (limited to 'tensorflow/go/genop')
-rw-r--r-- | tensorflow/go/genop/generate.sh | 5 | ||||
-rw-r--r-- | tensorflow/go/genop/main.go | 2 |
2 files changed, 6 insertions, 1 deletions
diff --git a/tensorflow/go/genop/generate.sh b/tensorflow/go/genop/generate.sh index 01fcfb9058..a894c87c27 100644 --- a/tensorflow/go/genop/generate.sh +++ b/tensorflow/go/genop/generate.sh @@ -19,6 +19,11 @@ 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" diff --git a/tensorflow/go/genop/main.go b/tensorflow/go/genop/main.go index 4d4940edf9..4a53084ed1 100644 --- a/tensorflow/go/genop/main.go +++ b/tensorflow/go/genop/main.go @@ -14,7 +14,7 @@ See the License for the specific language governing permissions and limitations under the License. */ -//go:generate sh generate.sh +//go:generate bash generate.sh // Command genop generates a Go source file with functions for TensorFlow ops. package main |