aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/common_runtime/direct_session.cc
diff options
context:
space:
mode:
authorGravatar Shanqing Cai <cais@google.com>2017-05-18 08:22:15 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-05-18 08:26:27 -0700
commit12ac2f34fadc8802121382c64588d9f9c2f58390 (patch)
tree1ca35f63dc579a8ac5e97fd6c98224d0d0e05935 /tensorflow/core/common_runtime/direct_session.cc
parentfaf55ee67c1efc25507402820cb7af6b2b57c446 (diff)
tfdbg: send encoded GraphDef Event protos in chunks to work around the grpc 4MB size limit
PiperOrigin-RevId: 156433576
Diffstat (limited to 'tensorflow/core/common_runtime/direct_session.cc')
-rw-r--r--tensorflow/core/common_runtime/direct_session.cc2
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/common_runtime/direct_session.cc b/tensorflow/core/common_runtime/direct_session.cc
index f1c17d7788..6a948b4ca7 100644
--- a/tensorflow/core/common_runtime/direct_session.cc
+++ b/tensorflow/core/common_runtime/direct_session.cc
@@ -391,7 +391,7 @@ Status DirectSession::DecorateAndPublishGraphForDebug(
DebugGraphDecoratorRegistry::CreateDecorator(debug_options, &decorator));
TF_RETURN_IF_ERROR(decorator->DecorateGraph(graph, device));
- TF_RETURN_IF_ERROR(decorator->PublishGraph(*graph));
+ TF_RETURN_IF_ERROR(decorator->PublishGraph(*graph, device->name()));
return Status::OK();
}