diff options
author | Asim Shankar <ashankar@google.com> | 2016-09-23 10:25:02 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-09-23 11:33:41 -0700 |
commit | 7bffed6ba12a4d463a0dfb1978e2108343c63eaf (patch) | |
tree | 292e668023177bb637d1fae08f1905b843f3a32e /tensorflow/go/session_test.go | |
parent | f982313d3c32cf2c7018eae9ec49759ae111d407 (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.go | 20 |
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) |