aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python
diff options
context:
space:
mode:
authorGravatar Skye Wanderman-Milne <skyewm@google.com>2018-06-26 13:57:36 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-06-26 13:59:58 -0700
commit1081683bf67f353dacc34c220c808a0080281f7f (patch)
tree1f7d053aa01341afae8f0e07eb107e9e4839d4fe /tensorflow/python
parentd0bd7f0a91ac3deaf93d56f7f8c89f98a1e34225 (diff)
Remove dead code from gradients_impl.py and gradients_test.py.
PiperOrigin-RevId: 202188895
Diffstat (limited to 'tensorflow/python')
-rw-r--r--tensorflow/python/ops/gradients_impl.py26
-rw-r--r--tensorflow/python/ops/gradients_test.py82
2 files changed, 0 insertions, 108 deletions
diff --git a/tensorflow/python/ops/gradients_impl.py b/tensorflow/python/ops/gradients_impl.py
index 250b9285c9..889a00190e 100644
--- a/tensorflow/python/ops/gradients_impl.py
+++ b/tensorflow/python/ops/gradients_impl.py
@@ -131,32 +131,6 @@ def _MarkReachedOps(from_ops, reached_ops):
queue.extend(output.consumers())
-def _GatherInputs(to_ops, reached_ops):
- """List all inputs of to_ops that are in reached_ops.
-
- Args:
- to_ops: list of Operations.
- reached_ops: set of Operations.
-
- Returns:
- The list of all inputs of to_ops that are in reached_ops.
- That list includes all elements of to_ops.
- """
- inputs = []
- queue = collections.deque()
- queue.extend(to_ops)
- while queue:
- op = queue.popleft()
- # We are interested in this op.
- if op in reached_ops:
- inputs.append(op)
- # Clear the boolean so we won't add the inputs again.
- reached_ops.remove(op)
- for inp in op.inputs:
- queue.append(inp.op)
- return inputs
-
-
def _PendingCount(to_ops, from_ops, colocate_gradients_with_ops):
"""Initialize the pending count for ops between two lists of Operations.
diff --git a/tensorflow/python/ops/gradients_test.py b/tensorflow/python/ops/gradients_test.py
index d81c756f1c..d70cd088c9 100644
--- a/tensorflow/python/ops/gradients_test.py
+++ b/tensorflow/python/ops/gradients_test.py
@@ -57,90 +57,8 @@ from tensorflow.python.ops.nn_ops import bias_add
from tensorflow.python.platform import googletest
-def _OpsBetween(to_ops, from_ops):
- """Build the list of operations between two lists of Operations.
-
- Args:
- to_ops: list of Operations.
- from_ops: list of Operations.
-
- Returns:
- The list of operations between "from_ops" and "to_ops", sorted by
- decreasing operation id. This list contains all elements of to_ops.
-
- TODO(touts): Think about returning an empty list if from_ops are not
- reachable from to_ops. Presently it returns to_ops in that case.
- """
- # Ops that are reachable from the output of "input_ops".
- reached_ops = set()
- # We only care to reach up to "output_ops" so we mark the
- # output ops as reached to avoid recursing past them.
- for op in to_ops:
- reached_ops.add(op)
- gradients_impl._MarkReachedOps(from_ops, reached_ops)
- between_ops = gradients_impl._GatherInputs(to_ops, reached_ops)
- between_ops.sort(key=lambda x: -x._id)
- return between_ops
-
-
class GradientsTest(test_util.TensorFlowTestCase):
- def _OpNames(self, op_list):
- return ["%s/%d" % (str(op.name), op._id) for op in op_list]
-
- def _assertOpListEqual(self, ops1, ops2):
- self.assertEquals(self._OpNames(ops1), self._OpNames(ops2))
-
- def testOpsBetweenSimple(self):
- with ops.Graph().as_default():
- t1 = constant(1.0)
- t2 = constant(2.0)
- t3 = array_ops.stack([t1, t2])
- # Full graph
- self._assertOpListEqual([t3.op, t2.op, t1.op],
- _OpsBetween([t3.op], [t1.op, t2.op]))
- # Only t1, t3.
- self._assertOpListEqual([t3.op, t1.op], _OpsBetween([t3.op], [t1.op]))
-
- def testOpsBetweenUnreachable(self):
- with ops.Graph().as_default():
- t1 = constant(1.0)
- t2 = constant(2.0)
- _ = array_ops.stack([t1, t2])
- t4 = constant(1.0)
- t5 = constant(2.0)
- t6 = array_ops.stack([t4, t5])
- # Elements of to_ops are always listed.
- self._assertOpListEqual([t6.op], _OpsBetween([t6.op], [t1.op]))
-
- def testOpsBetweenCut(self):
- with ops.Graph().as_default():
- t1 = constant(1.0)
- t2 = constant(2.0)
- t3 = array_ops.stack([t1, t2])
- t4 = constant([1.0])
- t5 = array_ops.concat([t4, t3], 0)
- t6 = constant([2.0])
- t7 = array_ops.concat([t5, t6], 0)
- self._assertOpListEqual([t7.op, t5.op, t4.op],
- _OpsBetween([t7.op], [t4.op]))
-
- def testOpsBetweenCycle(self):
- with ops.Graph().as_default():
- t1 = constant(1.0)
- t2 = constant(2.0)
- t3 = array_ops.stack([t1, t2])
- t4 = array_ops.concat([t3, t3, t3], 0)
- t5 = constant([1.0])
- t6 = array_ops.concat([t4, t5], 0)
- t7 = array_ops.concat([t6, t3], 0)
- self._assertOpListEqual([t6.op, t4.op, t3.op],
- _OpsBetween([t6.op], [t3.op]))
- self._assertOpListEqual([t7.op, t6.op, t5.op, t4.op, t3.op, t1.op],
- _OpsBetween([t7.op], [t1.op, t5.op]))
- self._assertOpListEqual([t6.op, t5.op, t4.op, t3.op, t2.op],
- _OpsBetween([t6.op], [t2.op, t5.op]))
-
def testGradients(self):
with ops.Graph().as_default():
inp = constant(1.0, shape=[32, 100], name="in")