Searched refs:should_save_summary (Results 1 – 11 of 11) sorted by relevance
91 should_save_summary=None): argument95 self.should_save_summary = should_save_summary104 should_save_summary=None): argument106 should_save_summary)123 if self.extended.should_save_summary is None:125 self.extended.should_save_summary = True127 self.extended.should_save_summary = False394 self.assertEqual(context.should_save_summary,395 strategy.extended.should_save_summary)407 context.should_save_summary)
321 def should_save_summary(self): member in _WorkerContext323 return self._strategy.extended.should_save_summary
461 def should_save_summary(self): member in OneDeviceExtended
686 def should_save_summary(self): member in ParameterServerStrategyExtended
2865 def should_save_summary(self): member in StrategyExtendedV13525 def should_save_summary(self): member in _DefaultDistributionExtended
911 def should_save_summary(self): member in CollectiveAllReduceExtended
833 def should_save_summary(self): member in MirroredExtended
1376 def should_save_summary(self): member in TPUExtended
27 name: "should_save_summary"
436 should_save_summary): argument440 self.should_save_summary = should_save_summary449 should_save_summary=None): argument451 should_save_summary)459 MockStrategy(should_save_summary=True), None, None, None)480 MockStrategy(should_save_summary=False), None, None, None)
374 if worker_context.should_save_summary:386 if worker_context.should_save_summary: