diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-01-12 12:00:29 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-12 12:07:23 -0800 |
| commit | b9d3aff0595bc3740e855f7b6782011103e85aee (patch) | |
| tree | af83688c4edecd9a060685d38287a7e778371668 /tensorflow/core/grappler/costs/virtual_scheduler.cc | |
| parent | 115367291897c5895c14f94a0ea46dc0ac6cdeec (diff) | |
Add _HostSend and _HostRecv to grappler
PiperOrigin-RevId: 181774069
Diffstat (limited to 'tensorflow/core/grappler/costs/virtual_scheduler.cc')
| -rw-r--r-- | tensorflow/core/grappler/costs/virtual_scheduler.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/grappler/costs/virtual_scheduler.cc b/tensorflow/core/grappler/costs/virtual_scheduler.cc index 29b401304d..0af889f886 100644 --- a/tensorflow/core/grappler/costs/virtual_scheduler.cc +++ b/tensorflow/core/grappler/costs/virtual_scheduler.cc @@ -341,7 +341,7 @@ Status VirtualScheduler::Init() { // to _Recv as control dependency when creating GrapplerItem. std::unordered_map<string, const NodeDef*> name_to_send; for (const auto& node : graph.node()) { - if (node.op() == "_Send") { + if (IsSend(node)) { const auto& attr = node.attr(); name_to_send[attr.at("tensor_name").s()] = &node; } |