aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/go/session_test.go
diff options
context:
space:
mode:
authorGravatar Jonathan Hseu <jhseu@google.com>2016-08-23 12:29:10 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-08-23 13:32:55 -0700
commit31e0b3bef5fa5e2108f8923633022bbd29b3d287 (patch)
treedbb42f3f2f182ab0d70cb35466c67099f0459a8f /tensorflow/go/session_test.go
parentb73bad5a5aaae94bfc112bfec3adc5fea8c006df (diff)
Rename Port to Output in the Go API.
Change: 131089292
Diffstat (limited to 'tensorflow/go/session_test.go')
-rw-r--r--tensorflow/go/session_test.go18
1 files changed, 9 insertions, 9 deletions
diff --git a/tensorflow/go/session_test.go b/tensorflow/go/session_test.go
index 78f6bccfd6..bebe2ac4de 100644
--- a/tensorflow/go/session_test.go
+++ b/tensorflow/go/session_test.go
@@ -19,27 +19,27 @@ import (
"testing"
)
-func Placeholder(g *Graph, name string, dt DataType) (Port, error) {
+func Placeholder(g *Graph, name string, dt DataType) (Output, error) {
b := newOpBuilder(g, "Placeholder", name)
b.SetAttrType("dtype", dt)
op, err := b.Build()
if err != nil {
- return Port{}, err
+ return Output{}, err
}
- return Port{op, 0}, nil
+ return Output{op, 0}, nil
}
-func Neg(g *Graph, name string, port Port) (Port, error) {
+func Neg(g *Graph, name string, port Output) (Output, error) {
b := newOpBuilder(g, "Neg", name)
b.AddInput(port)
op, err := b.Build()
if err != nil {
- return Port{}, err
+ return Output{}, err
}
- return Port{op, 0}, nil
+ return Output{op, 0}, nil
}
-func createTestGraph(t *testing.T, dt DataType) (*Graph, Port, Port) {
+func createTestGraph(t *testing.T, dt DataType) (*Graph, Output, Output) {
g := NewGraph()
inp, err := Placeholder(g, "p1", dt)
if err != nil {
@@ -72,7 +72,7 @@ func TestSessionRunNeg(t *testing.T) {
if err != nil {
t.Fatalf("NewSession() for %v: %v", test.input, err)
}
- output, err := s.Run(map[Port]*Tensor{inp: t1}, []Port{out}, []*Operation{out.Op})
+ output, err := s.Run(map[Output]*Tensor{inp: t1}, []Output{out}, []*Operation{out.Op})
if err != nil {
t.Fatalf("Run() for %v: %v", test.input, err)
}
@@ -103,7 +103,7 @@ func TestConcurrency(t *testing.T) {
}
for i := 0; i < 100; i++ {
// Session may close before Run() starts, so we don't check the error.
- go s.Run(map[Port]*Tensor{inp: tensor}, []Port{out}, []*Operation{out.Op})
+ go s.Run(map[Output]*Tensor{inp: tensor}, []Output{out}, []*Operation{out.Op})
}
if err = s.Close(); err != nil {
t.Errorf("Close() 1: %v", err)