Go: Support setting shape valued attributes.

Fixes #6833
Change: 144752893

1 files changed, 15 insertions, 0 deletions

diff --git a/tensorflow/go/operation_test.go b/tensorflow/go/operation_test.go
index 8080515ee9..4c4c960448 100644
--- a/tensorflow/go/operation_test.go
+++ b/tensorflow/go/operation_test.go
@@ -81,6 +81,21 @@ func TestOperationOutputListSize(t *testing.T) {
 	}
 }
 
+func TestOperationShapeAttribute(t *testing.T) {
+	g := NewGraph()
+	_, err := g.AddOperation(OpSpec{
+		Type: "Placeholder",
+		Attrs: map[string]interface{}{
+			"dtype": Float,
+			"shape": MakeShape(-1, 3),
+		},
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	// If and when the API to get attributes is added, check that here.
+}
+
 func TestOutputShape(t *testing.T) {
 	graph := NewGraph()
 	testdata := []struct {