diff options
author | Asim Shankar <ashankar@google.com> | 2017-01-17 14:04:20 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-01-17 14:31:05 -0800 |
commit | f8d75baaf4c92e76837de6bb64adccf8127a21d6 (patch) | |
tree | d3328471307fc541a14a95e8878d5cd9d32f24ab /tensorflow/go/genop | |
parent | 0662eabf9d6d670bd9a741ea3a3eb0c9f0005850 (diff) |
Go: Support setting shape valued attributes.
Fixes #6833
Change: 144752893
Diffstat (limited to 'tensorflow/go/genop')
-rw-r--r-- | tensorflow/go/genop/internal/genop.go | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/go/genop/internal/genop.go b/tensorflow/go/genop/internal/genop.go index 75c111e957..d9ebec0f8c 100644 --- a/tensorflow/go/genop/internal/genop.go +++ b/tensorflow/go/genop/internal/genop.go @@ -395,7 +395,7 @@ func goType(tfType string) (string, error) { case "type": gotype = "tf.DataType" case "shape": - gotype = "[]int64" + gotype = "tf.Shape" case "tensor": gotype = "tf.Tensor" case "string": |