diff options
Diffstat (limited to 'tensorflow/contrib/graph_editor/transform.py')
-rw-r--r-- | tensorflow/contrib/graph_editor/transform.py | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/tensorflow/contrib/graph_editor/transform.py b/tensorflow/contrib/graph_editor/transform.py index 26047437d7..11c19ccc22 100644 --- a/tensorflow/contrib/graph_editor/transform.py +++ b/tensorflow/contrib/graph_editor/transform.py @@ -446,10 +446,7 @@ class Transformer(object): # without any outputs. So the walk is now finalized from those roots. remaining_ops = [op for op in self._info.sgv.ops if op not in self._info.transformed_ops] - remaining_roots = [ - op for op in remaining_ops - if not op.outputs and not self._info.control_outputs.get(op) - ] + remaining_roots = [op for op in remaining_ops if not op.outputs] for op in remaining_roots: self._transform_op(op) |