Searched refs:_distribute_strategy (Results 1 – 14 of 14) sorted by relevance
40 self._distribute_strategy = strategy65 with ds_context.enter_or_assert_strategy(self._distribute_strategy):69 strategy=self._distribute_strategy,82 return self._distribute_strategy88 with ds_context.enter_or_assert_strategy(self._distribute_strategy):97 return self._distribute_strategy.extended.update(
452 self._distribute_strategy = strategy504 with ds_context.enter_or_assert_strategy(self._distribute_strategy):512 strategy=self._distribute_strategy,655 return self._distribute_strategy713 with ds_context.enter_or_assert_strategy(self._distribute_strategy):939 with ds_context.enter_or_assert_strategy(self._distribute_strategy):1156 with ds_context.enter_or_assert_strategy(self._distribute_strategy):1169 with ds_context.enter_or_assert_strategy(self._distribute_strategy):1182 with ds_context.enter_or_assert_strategy(self._distribute_strategy):1235 with ds_context.enter_or_assert_strategy(self._distribute_strategy):[all …]
225 variable_strategy = v._distribute_strategy # pylint: disable=protected-access
218 strategy = var._distribute_strategy # pylint: disable=protected-access
71 …distribute_strategy=distributed_variables[0]._distribute_strategy, # pylint: disable=protected-ac…
2151 …return v._distribute_strategy == self._container_strategy_weakref() # pylint: disable=protected-a…3423 return v._distribute_strategy is None # pylint: disable=protected-access
100 …strategy.experimental_local_results(var)[i]._distribute_strategy) # pylint: disable=protected-acc…
872 self.assertIs(strategy, created_step._distribute_strategy)
89 self.assertEqual(v._distribute_strategy, slot._distribute_strategy)
418 self._distribute_strategy = distribute_strategy505 distribute_strategy=self._distribute_strategy,910 distribute_strategy=self._distribute_strategy), ()
2634 def _distribute_strategy(self): member in RefVariable3039 def _distribute_strategy(self): member in PartitionedVariable
801 self._distribute_strategy = distribution_strategy_context.get_strategy()
83 def _distribute_strategy(self): member in TPUShardedVariable84 return self.variables[0]._distribute_strategy # pylint: disable=protected-access
52 self.assertIsNone(v._distribute_strategy)