Searched defs:should_checkpoint (Results 1 – 9 of 9) sorted by relevance
90 should_checkpoint=None, argument103 should_checkpoint=None, argument
316 def should_checkpoint(self): member in _WorkerContext
907 def should_checkpoint(self): member in CollectiveAllReduceExtended
457 def should_checkpoint(self): member in OneDeviceExtended
682 def should_checkpoint(self): member in ParameterServerStrategyExtended
829 def should_checkpoint(self): member in MirroredExtended
1372 def should_checkpoint(self): member in TPUExtended
2860 def should_checkpoint(self): member in StrategyExtendedV13521 def should_checkpoint(self): member in _DefaultDistributionExtended
435 def __init__(self, between_graph, should_init, should_checkpoint, argument448 should_checkpoint=None, argument