aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/graph/graph.cc
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2016-12-05 10:50:58 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-12-05 11:06:55 -0800
commit8a5dae396113655db2e4b16fe344dda4dca54994 (patch)
tree86024e08cf159e88d020017655bbe6006fc50f65 /tensorflow/core/graph/graph.cc
parent1e8ce51f43f5a3811740520d5a6205003e9665c2 (diff)
Adds VariableV2 with a sane shape_fn.
Change: 141071094
Diffstat (limited to 'tensorflow/core/graph/graph.cc')
-rw-r--r--tensorflow/core/graph/graph.cc1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/graph/graph.cc b/tensorflow/core/graph/graph.cc
index 648f439607..3e75d4435b 100644
--- a/tensorflow/core/graph/graph.cc
+++ b/tensorflow/core/graph/graph.cc
@@ -93,6 +93,7 @@ void Node::Initialize(int id, int cost_id, Properties* props) {
SET_CLASS(NC_HOST_RECV, ts, "_HostRecv", "");
SET_CLASS(NC_CONSTANT, ts, "Const", "HostConst");
SET_CLASS(NC_VARIABLE, ts, "Variable", "");
+ SET_CLASS(NC_VARIABLE, ts, "VariableV2", "");
SET_CLASS(NC_IDENTITY, ts, "Identity", "RefIdentity");
SET_CLASS(NC_GET_SESSION_HANDLE, ts, "GetSessionHandle", "");
SET_CLASS(NC_GET_SESSION_TENSOR, ts, "GetSessionTensor", "");