Searched refs:distribute_strategy (Results 1 – 5 of 5) sorted by relevance
2506 distribute_strategy = kwargs.get("distribute_strategy", None)2512 import_scope=import_scope, distribute_strategy=distribute_strategy)2534 distribute_strategy = kwargs.get("distribute_strategy", None)2545 import_scope=import_scope, distribute_strategy=distribute_strategy)
354 distribute_strategy=None): argument405 self._distribute_strategy = distribute_strategy722 distribute_strategy=self._distribute_strategy)1070 distribute_strategy=self._distribute_strategy), ()
552 def distribute_strategy(self): member in DistributedVariable834 def distribute_strategy(self): member in TPUMirroredVariable1471 def distribute_strategy(self): member in AggregatingVariable
319 self.assertIs(strategy, var.distribute_strategy)793 self.assertIs(distribution, mirrored.distribute_strategy)794 self.assertIs(distribution, sync_on_read.distribute_strategy)
888 self.assertIs(strategy, created_step.distribute_strategy)