diff options
| author |  Sami Kama <skama@nvidia.com> | 2018-06-08 20:05:37 -0700 |
|---|---|---|
| committer | Sami Kama <skama@nvidia.com> | 2018-06-11 18:21:07 -0700 |
| commit | d1120e1334aae84bff40b3ee7cf0a3849936fe4b (patch) | |
| tree | a07df79428f30f3f2f838e0b57f901464bb27b19 | |
| parent | d5aaf3fa4a4851abc6a0e5600474f7674f1adb93 (diff) | |
* Use VLOG(1) instead of std::cout in remapper.cc
* Remove op_op_lib dependency from ScopedAllocator. This dependency is
already satisfied through core and causes a fatal for libraries that
uses meta_optimizer due to double registration.
| -rw-r--r-- | tensorflow/core/grappler/optimizers/BUILD | 1 | ||||
| -rw-r--r-- | tensorflow/core/grappler/optimizers/remapper.cc | 4 |
2 files changed, 2 insertions, 3 deletions
diff --git a/tensorflow/core/grappler/optimizers/BUILD b/tensorflow/core/grappler/optimizers/BUILD index 0e22d4add8..5ed73eec50 100644 --- a/tensorflow/core/grappler/optimizers/BUILD +++ b/tensorflow/core/grappler/optimizers/BUILD @@ -780,7 +780,6 @@ cc_library( "//tensorflow/core:lib", "//tensorflow/core:lib_internal", "//tensorflow/core:protos_all_cc", - "//tensorflow/core:scoped_allocator_ops_op_lib", "//tensorflow/core/grappler:grappler_item", "//tensorflow/core/grappler:op_types", "//tensorflow/core/grappler:utils", diff --git a/tensorflow/core/grappler/optimizers/remapper.cc b/tensorflow/core/grappler/optimizers/remapper.cc index efd870b118..622fb134a1 100644 --- a/tensorflow/core/grappler/optimizers/remapper.cc +++ b/tensorflow/core/grappler/optimizers/remapper.cc @@ -22,6 +22,7 @@ limitations under the License. #include "tensorflow/core/grappler/op_types.h" #include "tensorflow/core/grappler/optimizers/constant_folding.h" #include "tensorflow/core/grappler/utils.h" +#include "tensorflow/core/platform/logging.h" namespace tensorflow { namespace grappler { @@ -200,8 +201,7 @@ Status Remapper::Optimize(Cluster* /*cluster*/, const GrapplerItem& item, } } if (optimizable) { - std::cout << "Optimizing fused batch norm node " << node.DebugString() - << std::endl; + VLOG(1)<< "Optimizing fused batch norm node " << node.DebugString(); AddBatchNormNodes(optimized_graph, node); continue; } |