diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-01-31 09:53:17 -0800 |
|---|---|---|
| committer |  Michael Case <mikecase@google.com> | 2018-01-31 17:08:10 -0800 |
| commit | 16b9fc676be6f8aacf06977a7f9439a56ffccefa (patch) | |
| tree | de2890c9997059bffd67caf6d24d9adfe31d0aff /tensorflow/contrib/meta_graph_transform | |
| parent | 76989a191815bdd96390626db154676ac42b890d (diff) | |
Extending sparsify_gather to remove variables from the tensorflow summaries.
PiperOrigin-RevId: 184004859
Diffstat (limited to 'tensorflow/contrib/meta_graph_transform')
| -rw-r--r-- | tensorflow/contrib/meta_graph_transform/meta_graph_transform.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/contrib/meta_graph_transform/meta_graph_transform.py b/tensorflow/contrib/meta_graph_transform/meta_graph_transform.py index 2932ae1c8d..ff88b4fa84 100644 --- a/tensorflow/contrib/meta_graph_transform/meta_graph_transform.py +++ b/tensorflow/contrib/meta_graph_transform/meta_graph_transform.py @@ -171,7 +171,14 @@ def _clean_save_and_restore(graph_def, op, removed_op_names): shape_op_value_tensor.tensor_shape.dim[0].size = len(shapes) op.attr['dtypes'].list.type[:] = dtypes + if not name_op.attr['_output_shapes'].list.shape: + name_op.attr['_output_shapes'].list.shape.add() + name_op.attr['_output_shapes'].list.shape[0].dim.add() name_op.attr['_output_shapes'].list.shape[0].dim[0].size = len(names) + + if not shape_op.attr['_output_shapes'].list.shape: + shape_op.attr['_output_shapes'].list.shape.add() + shape_op.attr['_output_shapes'].list.shape[0].dim.add() shape_op.attr['_output_shapes'].list.shape[0].dim[0].size = len(shapes) |