HomeSort by relevance Sort by last modified time
    Searched refs:collection_key (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/contrib/meta_graph_transform/
meta_graph_transform.py 579 def _get_single_node_name_from_collection(meta_graph_def, collection_key):
581 if collection_key not in meta_graph_def.collection_def:
583 collection = meta_graph_def.collection_def[collection_key]
587 collection_key))
591 collection_key, collection.bytes_list))
595 def _get_all_node_names_from_collection(meta_graph_def, collection_key):
597 if collection_key not in meta_graph_def.collection_def:
599 collection = meta_graph_def.collection_def[collection_key]
603 collection_key))
607 def _get_all_protos_from_collection(meta_graph_def, collection_key)
    [all...]
  /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
dnn_linear_combined.py 113 def _get_embedding_variable(column, collection_key, input_layer_scope):
114 return ops.get_collection(collection_key,
118 def _extract_embedding_lr_multipliers(embedding_lr_multipliers, collection_key,
130 column, collection_key, input_layer_scope)
    [all...]
  /external/tensorflow/tensorflow/python/training/
basic_session_run_hooks.py 490 collection_key = ops.GraphKeys.SAVERS
491 savers = ops.get_collection(collection_key)
495 "or provide a saver or scaffold.".format(collection_key))
500 format(collection_key))
    [all...]
monitored_session.py 253 def get_or_default(arg_name, collection_key, default_constructor):
255 elements = ops.get_collection(collection_key)
261 'tf.Scaffold(%s=item to use)', collection_key,
266 ops.add_to_collection(collection_key, op)
    [all...]
saver.py     [all...]
  /external/tensorflow/tensorflow/python/framework/
meta_graph_test.py 576 def _restore_collections_predicate(collection_key):
577 return (collection_key in include_collection_keys and
578 collection_key not in omit_collection_keys)
    [all...]

Completed in 105 milliseconds