diff options
| author |  AG Ramesh <ag.ramesh@intel.com> | 2018-07-22 08:20:44 -0700 |
|---|---|---|
| committer |  GitHub <noreply@github.com> | 2018-07-22 08:20:44 -0700 |
| commit | 99b1fa253c0d46ca95bbdd4890d7879d2b9e8f5f (patch) | |
| tree | 57be0ae379b3eed05c3d2cb10a2e2944e2c1ae1c /tensorflow/core/graph | |
| parent | 6743cb8ab362025229c79163c597dcec5ee70402 (diff) | |
minor format changes
Diffstat (limited to 'tensorflow/core/graph')
| -rw-r--r-- | tensorflow/core/graph/mkl_layout_pass.cc | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tensorflow/core/graph/mkl_layout_pass.cc b/tensorflow/core/graph/mkl_layout_pass.cc index ab427c0e83..3c94fa0cbb 100644 --- a/tensorflow/core/graph/mkl_layout_pass.cc +++ b/tensorflow/core/graph/mkl_layout_pass.cc @@ -2893,7 +2893,8 @@ class MklLayoutRewritePass : public GraphOptimizationPass { for (const Edge* e : n->in_edges()) { if (e->dst()->type_string() == csinfo_.lrn_grad && e->dst_input() == 2 && - e->src()->type_string() == mkl_op_registry::GetMklOpName(csinfo_.lrn) && e->src_output() == 0) { + e->src()->type_string() == mkl_op_registry::GetMklOpName(csinfo_.lrn) && + e->src_output() == 0) { do_rewrite = true; break; } @@ -2905,8 +2906,8 @@ class MklLayoutRewritePass : public GraphOptimizationPass { CHECK_NOTNULL(n); bool do_rewrite = false; for (const Edge* e : n->in_edges()) { - if (e->dst()->type_string() == csinfo_.max_pool_grad && - e->dst_input() == 1 && e->src()->type_string() == mkl_op_registry::GetMklOpName(csinfo_.max_pool) && + if (e->dst()->type_string() == csinfo_.max_pool_grad && e->dst_input() == 1 && + e->src()->type_string() == mkl_op_registry::GetMklOpName(csinfo_.max_pool) && e->src_output() == 0) { do_rewrite = true; break; |