diff options
author | Skye Wanderman-Milne <skyewm@google.com> | 2018-06-05 08:13:07 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-05 08:15:52 -0700 |
commit | 274f9510f68f237589df5c6a414e4b8e5ebcdba1 (patch) | |
tree | b779bc166e890b31c03f6f3e82987daf23070096 /tensorflow/contrib/graph_editor | |
parent | c0dc76a3994c743151404b1401599fefb9f37dd4 (diff) |
Remove _USE_C_API staging from ops.py.
PiperOrigin-RevId: 199298594
Diffstat (limited to 'tensorflow/contrib/graph_editor')
-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 592d37b432..026a3d1200 100644 --- a/tensorflow/contrib/graph_editor/transform.py +++ b/tensorflow/contrib/graph_editor/transform.py @@ -189,9 +189,6 @@ def copy_op_handler(info, op, new_inputs, copy_shape=True, nodedef_fn=None): if op._original_op: op_._original_op = op._original_op - # Add op to the graph - info.graph_._add_op(op_) - return op_, op_.outputs @@ -492,7 +489,7 @@ class Transformer(object): t_ = info.transformed_ts[t] consumer_op_ = info.transformed_ops[consumer_op] t_index_ = list(consumer_op_.inputs).index(tmp_t_) - consumer_op_._update_input(t_index_, t_, update_dtype=False) # pylint: disable=protected-access + consumer_op_._update_input(t_index_, t_) # pylint: disable=protected-access def _connect_control_inputs(self, info): """Connect the previously copied ops.""" |