| /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/ |
| export.py | 56 saver = _get_first_op_from_collection(ops.GraphKeys.SAVERS)
64 ops.add_to_collection(ops.GraphKeys.SAVERS, saver)
|
| /external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
| meta_graph_transform.py | 408 # Primary SaverDef and SAVERS collection
413 base_meta_graph_def, _ops.GraphKeys.SAVERS))
415 savers = initializer_names.get(_ops.GraphKeys.SAVERS, [])
416 savers.extend([
420 initializer_names[_ops.GraphKeys.SAVERS] = savers
|
| /external/tensorflow/tensorflow/python/tpu/ |
| async_checkpoint.py | 194 # Get saver from the SAVERS collection if present.
195 collection_key = ops.GraphKeys.SAVERS
196 savers = ops.get_collection(collection_key)
197 if not savers:
201 elif len(savers) > 1:
207 self._saver = savers[0]
208 return savers[0]
|
| session_support.py | 297 is None the `SAVERS` collection will be read to find a saver.
350 savers = ops.get_collection(ops.GraphKeys.SAVERS)
351 if not savers:
354 if not isinstance(savers, list):
355 return savers
357 if len(savers) > 1:
359 'Multiple savers in the SAVERS collection. On-demand checkpointing '
364 return savers[0
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| graph_actions.py | 422 saver = _get_first_op_from_collection(ops.GraphKeys.SAVERS)
425 ops.add_to_collection(ops.GraphKeys.SAVERS, saver)
|
| graph_actions_test.py | 380 ops.add_to_collection(ops.GraphKeys.SAVERS, saver_lib.Saver())
|
| /external/tensorflow/tensorflow/python/training/ |
| supervisor.py | 468 saver = self._get_first_op_from_collection(ops.GraphKeys.SAVERS)
471 ops.add_to_collection(ops.GraphKeys.SAVERS, saver)
[all...] |
| saver.py | 72 """Base class for Savers.
583 """Returns the saver from SAVERS collection, or creates a default one.
592 RuntimeError: If the SAVERS collection already has more than one items.
594 collection_key = ops.GraphKeys.SAVERS
595 savers = ops.get_collection(collection_key)
596 if savers:
597 if len(savers) > 1:
602 return savers[0]
623 Savers can automatically number checkpoint filenames with a provided counter.
626 step number. To avoid filling up disks, savers manage checkpoint file
[all...] |
| basic_session_run_hooks.py | 620 # Get saver from the SAVERS collection if present.
621 collection_key = ops.GraphKeys.SAVERS
622 savers = ops.get_collection(collection_key)
623 if not savers:
627 elif len(savers) > 1:
633 self._saver = savers[0]
634 return savers[0]
[all...] |
| saver_test.py | [all...] |
| monitored_session_test.py | 125 ops.add_to_collection(ops.GraphKeys.SAVERS, saver_lib.Saver())
126 ops.add_to_collection(ops.GraphKeys.SAVERS, saver_lib.Saver())
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| estimator.py | [all...] |
| /external/tensorflow/tensorflow/python/framework/ |
| meta_graph.py | 293 """Identifies any nodes in the graph_def related to unused Savers.
296 scope, so we need only identify the scopes associated with extraneous Savers
582 clear_extraneous_savers: Remove any preexisting SaverDefs from the SAVERS
652 if clear_extraneous_savers and ctype == ops.GraphKeys.SAVERS:
[all...] |
| ops.py | [all...] |
