Searched refs:_experimental_distribute_values_from_function (Results 1 – 6 of 6) sorted by relevance
1753 …return self._extended._experimental_distribute_values_from_function( # pylint: disable=protected-…2293 def _experimental_distribute_values_from_function(self, value_fn): member in StrategyExtendedV23660 def _experimental_distribute_values_from_function(self, value_fn): member in _DefaultDistributionExtended
341 def _experimental_distribute_values_from_function(self, value_fn): member in OneDeviceExtended
400 def _experimental_distribute_values_from_function(self, value_fn): member in ParameterServerStrategyExtended
597 def _experimental_distribute_values_from_function(self, value_fn): member in MirroredExtended
683 def _experimental_distribute_values_from_function(self, value_fn): member in CollectiveAllReduceExtended
1037 def _experimental_distribute_values_from_function(self, value_fn): member in TPUExtended