| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_coordinator_test.py | 87 should_checkpoint=None, argument
91 self.should_checkpoint = should_checkpoint
100 should_checkpoint=None, argument
102 self.extended = MockExtended(between_graph, should_init, should_checkpoint,
115 if self.extended.should_checkpoint is None:
117 self.extended.should_checkpoint = True
119 self.extended.should_checkpoint = False
373 self.assertEqual(context.should_checkpoint,
374 strategy.extended.should_checkpoint)
387 context.should_checkpoint,
|
| D | distribute_coordinator.py | 316 def should_checkpoint(self): member in _WorkerContext
318 return self._strategy.extended.should_checkpoint
|
| D | one_device_strategy.py | 195 def should_checkpoint(self): member in OneDeviceExtended
|
| D | collective_all_reduce_strategy.py | 462 def should_checkpoint(self): member in CollectiveAllReduceExtended
|
| D | parameter_server_strategy.py | 541 def should_checkpoint(self): member in ParameterServerStrategyExtended
|
| D | tpu_strategy.py | 566 def should_checkpoint(self): member in TPUExtended
|
| D | distribute_lib.py | 1348 def should_checkpoint(self): member in DistributionStrategyExtended
|
| D | mirrored_strategy.py | 777 def should_checkpoint(self): member in MirroredExtended
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 22 name: "should_checkpoint"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 22 name: "should_checkpoint"
|
| /external/tensorflow/tensorflow/python/training/ |
| D | monitored_session_test.py | 389 def __init__(self, between_graph, should_init, should_checkpoint, argument
393 self.should_checkpoint = should_checkpoint
402 should_checkpoint=None, argument
404 self.extended = MockExtended(between_graph, should_init, should_checkpoint,
454 MockStrategy(should_checkpoint=True), None, None, None)
474 MockStrategy(should_checkpoint=False), None, None, None)
|
| D | monitored_session.py | 387 if worker_context.should_checkpoint:
|