| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | distributed_file_utils.py | 92 if strategy.extended.should_checkpoint:
116 not strategy.extended.should_checkpoint):
|
| D | worker_training_state.py | 79 if self._model.distribute_strategy.extended.should_checkpoint:
|
| D | distribute_coordinator_utils.py | 277 def should_checkpoint(self): member in _WorkerContext
279 return self._strategy.extended.should_checkpoint
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_coordinator_test.py | 79 should_checkpoint=None, argument
83 self.should_checkpoint = should_checkpoint
92 should_checkpoint=None, argument
94 self.extended = MockExtended(between_graph, should_init, should_checkpoint,
107 if self.extended.should_checkpoint is None:
109 self.extended.should_checkpoint = True
111 self.extended.should_checkpoint = False
378 self.assertEqual(context.should_checkpoint,
379 strategy.extended.should_checkpoint)
392 context.should_checkpoint,
|
| D | multi_worker_util.py | 279 return dc_context.get_current_worker_context().should_checkpoint
|
| D | distribute_coordinator.py | 312 def should_checkpoint(self): member in _WorkerContext
314 return self._strategy.extended.should_checkpoint
|
| D | one_device_strategy.py | 457 def should_checkpoint(self): member in OneDeviceExtended
|
| D | parameter_server_strategy.py | 669 def should_checkpoint(self): member in ParameterServerStrategyExtended
|
| D | distribute_lib.py | 2981 def should_checkpoint(self): member in StrategyExtendedV1
3750 def should_checkpoint(self): member in _DefaultDistributionExtended
|
| D | collective_all_reduce_strategy.py | 962 def should_checkpoint(self): member in CollectiveAllReduceExtended
|
| D | mirrored_strategy.py | 875 def should_checkpoint(self): member in MirroredExtended
|
| D | tpu_strategy.py | 1497 def should_checkpoint(self): member in TPUExtended
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 23 name: "should_checkpoint"
|
| /external/tensorflow/tensorflow/python/training/ |
| D | monitored_session_test.py | 472 def __init__(self, between_graph, should_init, should_checkpoint, argument
476 self.should_checkpoint = should_checkpoint
485 should_checkpoint=None, argument
487 self.extended = MockExtended(between_graph, should_init, should_checkpoint,
537 MockStrategy(should_checkpoint=True), None, None, None)
557 MockStrategy(should_checkpoint=False), None, None, None)
|
| D | monitored_session.py | 398 if worker_context.should_checkpoint:
|