diff options
| author |  Yao Zhang <yaozhang@google.com> | 2017-08-28 23:24:45 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-08-28 23:29:00 -0700 |
| commit | a0cb8c528604819724059da1bbfbf8f2b95fda98 (patch) | |
| tree | a67407265e4fc470f5193b5cc466691fd442f7ce /tensorflow/core/grappler/utils.cc | |
| parent | c32d401c51ce004dcd33811e7c9027bf259f38a0 (diff) | |
Support loop for layout optimizer.
PiperOrigin-RevId: 166808322
Diffstat (limited to 'tensorflow/core/grappler/utils.cc')
| -rw-r--r-- | tensorflow/core/grappler/utils.cc | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/utils.cc b/tensorflow/core/grappler/utils.cc index 948df18879..9e15744fab 100644 --- a/tensorflow/core/grappler/utils.cc +++ b/tensorflow/core/grappler/utils.cc @@ -177,5 +177,13 @@ bool ExecuteWithTimeout(std::function<void()> fn, const int64 timeout_in_ms, return notified; } +string AsControlDependency(const NodeDef& node) { + return strings::StrCat("^", node.name()); +} + +string AsControlDependency(const string& node) { + return strings::StrCat("^", node); +} + } // end namespace grappler } // end namespace tensorflow |