aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/go/session_test.go
diff options
context:
space:
mode:
authorGravatar Asim Shankar <ashankar@google.com>2016-09-23 10:25:02 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-09-23 11:33:41 -0700
commit7bffed6ba12a4d463a0dfb1978e2108343c63eaf (patch)
tree292e668023177bb637d1fae08f1905b843f3a32e /tensorflow/go/session_test.go
parentf982313d3c32cf2c7018eae9ec49759ae111d407 (diff)
go: Ability to import a pre-defined Graph.
With this change, it should be possible to execute a pre-defined Graph created by any means (like a training session in a Python program) in Go. One more step towards #10 Change: 134096795
Diffstat (limited to 'tensorflow/go/session_test.go')
-rw-r--r--tensorflow/go/session_test.go20
1 files changed, 0 insertions, 20 deletions
diff --git a/tensorflow/go/session_test.go b/tensorflow/go/session_test.go
index bebe2ac4de..6e823d1841 100644
--- a/tensorflow/go/session_test.go
+++ b/tensorflow/go/session_test.go
@@ -19,26 +19,6 @@ import (
"testing"
)
-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 Output{}, err
- }
- return Output{op, 0}, nil
-}
-
-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 Output{}, err
- }
- return Output{op, 0}, nil
-}
-
func createTestGraph(t *testing.T, dt DataType) (*Graph, Output, Output) {
g := NewGraph()
inp, err := Placeholder(g, "p1", dt)