Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/tpu/
Dasync_checkpoint.py205 collection_key = ops.GraphKeys.SAVERS
206 savers = ops.get_collection(collection_key)
210 "or provide a saver or scaffold.".format(collection_key))
215 .format(collection_key))
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py553 for collection_key in self._collections_to_keep(
555 for op in sess.graph.get_collection(collection_key):
568 for collection_key in self._collections_to_keep(
570 self._grappler_meta_graph_def.collection_def[collection_key].CopyFrom(
571 input_meta_graph_def.collection_def[collection_key])
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.-scaffold.pbtxt55 …argspec: "args=[\'arg_name\', \'collection_key\', \'default_constructor\'], varargs=None, keywords…
/external/tensorflow/tensorflow/python/training/
Dbasic_session_run_hooks.py643 collection_key = ops.GraphKeys.SAVERS
644 savers = ops.get_collection(collection_key)
648 "or provide a saver or scaffold.".format(collection_key))
653 .format(collection_key))
Dmonitored_session.py286 def get_or_default(arg_name, collection_key, default_constructor): argument
288 elements = ops.get_collection(collection_key)
295 'tf.Scaffold(%s=item to use)', collection_key, arg_name)
299 ops.add_to_collection(collection_key, op)
Dsaver.py598 collection_key = ops.GraphKeys.SAVERS
599 savers = ops.get_collection(collection_key)
605 .format(collection_key))
609 ops.add_to_collection(collection_key, saver)
/external/tensorflow/tensorflow/python/framework/
Dmeta_graph_test.py672 def _restore_collections_predicate(collection_key): argument
673 return (collection_key in include_collection_keys and
674 collection_key not in omit_collection_keys)
Dtest_util.py696 for collection_key in ops.get_default_graph().collections:
697 collection = ops.get_collection(collection_key)
698 size_before = collection_sizes_before.get(collection_key, 0)
703 (collection_key, size_before, len(collection), collection))
707 del collection_key