diff options
author | Geoffrey Irving <geoffreyi@google.com> | 2017-06-01 15:24:16 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-06-01 15:28:43 -0700 |
commit | 8032e1f75dd6a56b39a07890f60acc6b275d0683 (patch) | |
tree | 846f20b2be63f8d28802f37a1f408a6779042098 /tensorflow/core/common_runtime/function_test.cc | |
parent | 2e44be35dc037b9c191569fd43caf1a7fcfceaec (diff) |
Make function instantiation use std::vector<NodeDef> instead of GraphDef
It's about to turn into std::vector<NodeInfoPtr>; this change gets us partway there.
RELNOTES: n/a
PiperOrigin-RevId: 157771141
Diffstat (limited to 'tensorflow/core/common_runtime/function_test.cc')
-rw-r--r-- | tensorflow/core/common_runtime/function_test.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/common_runtime/function_test.cc b/tensorflow/core/common_runtime/function_test.cc index e27fc3898d..dec6ca996a 100644 --- a/tensorflow/core/common_runtime/function_test.cc +++ b/tensorflow/core/common_runtime/function_test.cc @@ -93,7 +93,7 @@ class FunctionTest : public ::testing::Test { GraphConstructorOptions opts; opts.allow_internal_ops = true; opts.expect_device_spec = false; - TF_CHECK_OK(ConvertGraphDefToGraph(opts, result.gdef, g)); + TF_CHECK_OK(ConvertNodeDefsToGraph(opts, result.nodes, g)); const int version = g->versions().producer(); LocalExecutorParams params; @@ -949,7 +949,7 @@ GraphDef Optimize(const std::function<bool(Graph* g)>& pass, GraphConstructorOptions opts; opts.allow_internal_ops = true; opts.expect_device_spec = false; - TF_CHECK_OK(ConvertGraphDefToGraph(opts, result.gdef, g.get())); + TF_CHECK_OK(ConvertNodeDefsToGraph(opts, result.nodes, g.get())); pass(g.get()); std::unique_ptr<Graph> g1(new Graph(OpRegistry::Global())); CopyGraph(*g, g1.get()); |