diff options
author | Jason Taylor <jason.taylor@algolux.com> | 2018-06-14 16:18:27 -0400 |
---|---|---|
committer | Jason Taylor <jason.taylor@algolux.com> | 2018-06-14 16:18:27 -0400 |
commit | bc3204f3b030ba84120d46bb8503c615624448e9 (patch) | |
tree | 53d73886c3ec18b0833fd60d1225665083816825 /tensorflow/contrib/graph_editor | |
parent | c75734ab7f0ba0aef608738e18ce73847d16958a (diff) |
tf.contrib.graph_editor.transform.copy_op_handler: change default copy_shape to False so graph_replace can work for network pruning where new weight shape does not match old weight shape
Diffstat (limited to 'tensorflow/contrib/graph_editor')
-rw-r--r-- | tensorflow/contrib/graph_editor/transform.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/graph_editor/transform.py b/tensorflow/contrib/graph_editor/transform.py index 026a3d1200..e79ccd8da1 100644 --- a/tensorflow/contrib/graph_editor/transform.py +++ b/tensorflow/contrib/graph_editor/transform.py @@ -129,7 +129,7 @@ def transform_op_if_inside_handler(info, op, keep_if_possible=True): return None -def copy_op_handler(info, op, new_inputs, copy_shape=True, nodedef_fn=None): +def copy_op_handler(info, op, new_inputs, copy_shape=False, nodedef_fn=None): """Copy a `tf.Operation`. Args: |