Searched defs:should_checkpoint (Results 1 – 10 of 10) sorted by relevance
90 should_checkpoint=None, argument103 should_checkpoint=None, argument
316 def should_checkpoint(self): member in _WorkerContext
459 def should_checkpoint(self): member in OneDeviceExtended
924 def should_checkpoint(self): member in CollectiveAllReduceExtended
671 def should_checkpoint(self): member in ParameterServerStrategyExtended
869 def should_checkpoint(self): member in MirroredExtended
1382 def should_checkpoint(self): member in TPUExtended
2974 def should_checkpoint(self): member in StrategyExtendedV13745 def should_checkpoint(self): member in _DefaultDistributionExtended
281 def should_checkpoint(self): member in _WorkerContext
476 def __init__(self, between_graph, should_init, should_checkpoint, argument489 should_checkpoint=None, argument