Searched defs:should_checkpoint (Results 1 – 10 of 10) sorted by relevance
79 should_checkpoint=None, argument92 should_checkpoint=None, argument
312 def should_checkpoint(self): member in _WorkerContext
962 def should_checkpoint(self): member in CollectiveAllReduceExtended
457 def should_checkpoint(self): member in OneDeviceExtended
669 def should_checkpoint(self): member in ParameterServerStrategyExtended
875 def should_checkpoint(self): member in MirroredExtended
2981 def should_checkpoint(self): member in StrategyExtendedV13750 def should_checkpoint(self): member in _DefaultDistributionExtended
1497 def should_checkpoint(self): member in TPUExtended
277 def should_checkpoint(self): member in _WorkerContext
472 def __init__(self, between_graph, should_init, should_checkpoint, argument485 should_checkpoint=None, argument