Searched refs:experimental_should_init (Results 1 – 12 of 12) sorted by relevance
90 self.experimental_should_init = should_init110 if self.extended.experimental_should_init is None:112 self.extended.experimental_should_init = True114 self.extended.experimental_should_init = False371 self.assertEqual(context._strategy.extended.experimental_should_init,372 strategy.extended.experimental_should_init)386 context._strategy.extended.experimental_should_init,
250 if not self._strategy or self._strategy.extended.experimental_should_init:311 def experimental_should_init(self): member in _WorkerContext313 return self._strategy.extended.experimental_should_init
191 def experimental_should_init(self): member in OneDeviceExtended
458 def experimental_should_init(self): member in CollectiveAllReduceExtended
537 def experimental_should_init(self): member in ParameterServerStrategyExtended
562 def experimental_should_init(self): member in TPUExtended
1343 def experimental_should_init(self): member in DistributionStrategyExtended
773 def experimental_should_init(self): member in MirroredExtended
14 name: "experimental_should_init"
347 if not worker_context or worker_context.experimental_should_init:
392 self.experimental_should_init = should_init