diff options
| author |  Martin Wicke <wicke@google.com> | 2017-03-23 12:31:16 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-03-23 13:44:29 -0700 |
| commit | bc456e361d49d1d89a74b80060c70efb51fd7d87 (patch) | |
| tree | 825e04287f1e2d2ac098ca3f0fdd4e361aefd68c /tensorflow/tools/graph_transforms/summarize_graph_main.cc | |
| parent | 8ca071456537e6c96ae8896c2a20b1f08b0e59d3 (diff) | |
Merge changes from github.
Change: 151046259
Diffstat (limited to 'tensorflow/tools/graph_transforms/summarize_graph_main.cc')
| -rw-r--r-- | tensorflow/tools/graph_transforms/summarize_graph_main.cc | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/tools/graph_transforms/summarize_graph_main.cc b/tensorflow/tools/graph_transforms/summarize_graph_main.cc index 8c23ae7a74..f45dfbba0c 100644 --- a/tensorflow/tools/graph_transforms/summarize_graph_main.cc +++ b/tensorflow/tools/graph_transforms/summarize_graph_main.cc @@ -109,7 +109,7 @@ Status SummarizeGraph(const GraphDef& graph, const string& graph_path) { if (node.op() == "Placeholder") { placeholders.push_back(&node); } - if (node.op() == "Variable") { + if (node.op() == "Variable" || node.op() == "VariableV2") { variables.push_back(&node); } } @@ -168,7 +168,8 @@ Status SummarizeGraph(const GraphDef& graph, const string& graph_path) { if (node.device() != "") { ++device_counts[node.device()]; } - if ((node.op() == "Const") || (node.op() == "Variable")) { + if ((node.op() == "Const") || (node.op() == "Variable") || + (node.op() == "VariableV2")) { Tensor tensor; if (node.attr().count("value") && tensor.FromProto(node.attr().at("value").tensor())) { |