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

  /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/opt/python/training/
elastic_average_optimizer_test.py 71 savers = []
140 savers.append(saver)
142 return sessions, graphs, train_ops, savers
157 sessions, graphs, train_ops, savers = _get_workers(
201 savers[0].save(sessions[0]._sess._sess._sess._sess,
220 sessions, graphs, train_ops, savers = _get_workers(
255 savers[0].save(sessions[0]._sess._sess._sess._sess,
  /external/tensorflow/tensorflow/python/training/
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.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...]
saver_test.py     [all...]
  /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/core/grappler/
grappler_item_builder.cc 556 if (meta_graph.collection_def().count("savers") > 0) {
557 const CollectionDef& savers = meta_graph.collection_def().at("savers"); local
558 for (const auto& raw : savers.bytes_list().value()) {
560 // Skip bad savers since we don't need saves/restores to be able to run a

Completed in 205 milliseconds