aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/graph_editor
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2017-01-31 03:55:27 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-01-31 04:09:11 -0800
commit8c2b19889e49df88f3c4c0331d6930f887f5adae (patch)
treee1537383c203f0f00a0ddf53e673c89eb377c488 /tensorflow/contrib/graph_editor
parent2490fcade664f5dc5f6ef9b88fc7d5bb30cf738a (diff)
Added missing build targets.
Change logging.info -> logging.debug Change: 146103023
Diffstat (limited to 'tensorflow/contrib/graph_editor')
-rw-r--r--tensorflow/contrib/graph_editor/BUILD2
-rw-r--r--tensorflow/contrib/graph_editor/transform.py6
2 files changed, 5 insertions, 3 deletions
diff --git a/tensorflow/contrib/graph_editor/BUILD b/tensorflow/contrib/graph_editor/BUILD
index 22abeb0151..188d5abf2c 100644
--- a/tensorflow/contrib/graph_editor/BUILD
+++ b/tensorflow/contrib/graph_editor/BUILD
@@ -23,6 +23,8 @@ py_library(
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:framework_for_generated_wrappers",
+ "//tensorflow/python:platform",
+ "@six_archive//:six",
],
)
diff --git a/tensorflow/contrib/graph_editor/transform.py b/tensorflow/contrib/graph_editor/transform.py
index 4d272e108f..57fd7d7188 100644
--- a/tensorflow/contrib/graph_editor/transform.py
+++ b/tensorflow/contrib/graph_editor/transform.py
@@ -170,7 +170,7 @@ def copy_op_handler(info, op, copy_shape=True):
if op._original_op:
original_op = info.transform_original_op_handler(info, op._original_op)
if original_op is None:
- logging.info("Could not find original op of: %s", op_.name)
+ logging.debug("Could not find original op of: %s", op_.name)
else:
op_._original_op = original_op
@@ -442,7 +442,7 @@ class Transformer(object):
def _copy_ops(self, info):
"""Copy ops without connecting them."""
for op in info.sgv.ops:
- logging.info("Copying op: %s", op.name)
+ logging.debug("Copying op: %s", op.name)
# TODO(fkp): return a subgraph?
op_, op_outputs_ = self.transform_op_handler(info, op)
if op is op_:
@@ -460,7 +460,7 @@ class Transformer(object):
def _connect_ops(self, info):
"""Connect the previously copied ops."""
for op in info.sgv.ops:
- logging.info("Finalizing op: %s", op.name)
+ logging.debug("Finalizing op: %s", op.name)
op_ = info.transformed_ops[op]
# pylint: disable=protected-access