Searched refs:_distribute_strategy (Results 1 – 7 of 7) sorted by relevance
445 self._distribute_strategy = strategy553 return self._distribute_strategy577 return self._distribute_strategy.extended.read_var(self)591 variable_strategy = v._distribute_strategy # pylint: disable=protected-access664 with _enter_or_assert_strategy(self._distribute_strategy):676 return self._distribute_strategy.extended.update(679 _assert_replica_context(self._distribute_strategy)776 self._distribute_strategy = strategy835 return self._distribute_strategy934 with _enter_or_assert_strategy(self._distribute_strategy):[all …]
918 …return v._distribute_strategy == self._container_strategy_weakref() # pylint: disable=protected-a…1570 return v._distribute_strategy is None # pylint: disable=protected-access
405 self._distribute_strategy = distribute_strategy722 distribute_strategy=self._distribute_strategy)1070 distribute_strategy=self._distribute_strategy), ()
2213 def _distribute_strategy(self): member in RefVariable2610 def _distribute_strategy(self): member in PartitionedVariable
907 self.assertIs(strategy, created_step._distribute_strategy)
322 self.assertIs(strategy, var.get(d)._distribute_strategy) # pylint: disable=protected-access
49 self.assertIsNone(v._distribute_strategy)