Searched refs:should_save_summary (Results 1 – 12 of 12) sorted by relevance
88 should_save_summary=None): argument92 self.should_save_summary = should_save_summary101 should_save_summary=None): argument103 should_save_summary)120 if self.extended.should_save_summary is None:122 self.extended.should_save_summary = True124 self.extended.should_save_summary = False375 self.assertEqual(context.should_save_summary,376 strategy.extended.should_save_summary)388 context.should_save_summary)
321 def should_save_summary(self): member in _WorkerContext323 return self._strategy.extended.should_save_summary
199 def should_save_summary(self): member in OneDeviceExtended
466 def should_save_summary(self): member in CollectiveAllReduceExtended
545 def should_save_summary(self): member in ParameterServerStrategyExtended
570 def should_save_summary(self): member in TPUExtended
1353 def should_save_summary(self): member in DistributionStrategyExtended
781 def should_save_summary(self): member in MirroredExtended
26 name: "should_save_summary"
390 should_save_summary): argument394 self.should_save_summary = should_save_summary403 should_save_summary=None): argument405 should_save_summary)413 MockStrategy(should_save_summary=True), None, None, None)434 MockStrategy(should_save_summary=False), None, None, None)
357 if worker_context.should_save_summary:369 if worker_context.should_save_summary: