diff options
| author |  Sami Kama <samikama@users.noreply.github.com> | 2018-06-09 09:16:02 -0700 |
|---|---|---|
| committer |  drpngx <drpngx@users.noreply.github.com> | 2018-06-09 09:16:02 -0700 |
| commit | 14e7f42ae0ff488b83f00cccaf350aec1032af5c (patch) | |
| tree | d553675a3cc49fde1644c61a6037e0d50f308ed0 | |
| parent | a6a265b61a9ad9510f45cf4c9032778bf2e042b9 (diff) | |
* Use VLOG(1) instead of std::cout in remapper.cc (#19870)
* 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 20887bc218..2073c2968b 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; } |