experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix create_meta_graph to respect an empty collection_list.

Browse Source 
PiperOrigin-RevId: 158073112
This commit is contained in:
A. Unique TensorFlower 2017-06-05 15:47:20 -07:00 committed by TensorFlower Gardener
parent 54ccc3e5a8
commit 3d52e4cb93
1 changed files with 1 additions and 1 deletions

2
tensorflow/python/framework/meta_graph.py
View File

@ -510,7 +510,7 @@ def create_meta_graph_def(meta_info_def=None,
meta_graph_def.saver_def.MergeFrom(saver_def) meta_graph_def.saver_def.MergeFrom(saver_def)
# Adds collection_list. # Adds collection_list.
if collection_list: if collection_list is not None:
clist = collection_list clist = collection_list
else: else:
clist = graph.get_all_collection_keys() clist = graph.get_all_collection_keys()