Home
last modified time | relevance | path

Searched refs:collection_key (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/contrib/meta_graph_transform/
Dmeta_graph_transform.py589 def _get_single_node_name_from_collection(meta_graph_def, collection_key): argument
591 if collection_key not in meta_graph_def.collection_def:
593 collection = meta_graph_def.collection_def[collection_key]
597 collection_key))
601 collection_key, collection.bytes_list))
605 def _get_all_node_names_from_collection(meta_graph_def, collection_key): argument
607 if collection_key not in meta_graph_def.collection_def:
609 collection = meta_graph_def.collection_def[collection_key]
613 collection_key))
617 def _get_all_protos_from_collection(meta_graph_def, collection_key): argument
[all …]
/external/tensorflow/tensorflow/python/tpu/
Dasync_checkpoint.py195 collection_key = ops.GraphKeys.SAVERS
196 savers = ops.get_collection(collection_key)
200 "or provide a saver or scaffold.".format(collection_key))
205 .format(collection_key))
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Ddnn_linear_combined.py118 def _get_embedding_variable(column, collection_key, input_layer_scope): argument
119 return ops.get_collection(collection_key,
123 def _extract_embedding_lr_multipliers(embedding_lr_multipliers, collection_key, argument
135 column, collection_key, input_layer_scope)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.-scaffold.pbtxt51 …argspec: "args=[\'arg_name\', \'collection_key\', \'default_constructor\'], varargs=None, keywords…
/external/tensorflow/tensorflow/python/training/
Dbasic_session_run_hooks.py621 collection_key = ops.GraphKeys.SAVERS
622 savers = ops.get_collection(collection_key)
626 "or provide a saver or scaffold.".format(collection_key))
631 format(collection_key))
Dmonitored_session.py273 def get_or_default(arg_name, collection_key, default_constructor): argument
275 elements = ops.get_collection(collection_key)
282 'tf.Scaffold(%s=item to use)', collection_key, arg_name)
286 ops.add_to_collection(collection_key, op)
Dsaver.py594 collection_key = ops.GraphKeys.SAVERS
595 savers = ops.get_collection(collection_key)
601 format(collection_key))
605 ops.add_to_collection(collection_key, saver)
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py545 for collection_key in ops.get_default_graph().collections:
546 collection = ops.get_collection(collection_key)
547 size_before = collection_sizes_before.get(collection_key, 0)
552 (collection_key, size_before, len(collection), collection))
556 del collection_key
Dmeta_graph_test.py626 def _restore_collections_predicate(collection_key): argument
627 return (collection_key in include_collection_keys and
628 collection_key not in omit_collection_keys)