_experimental_distribute_values_from_function (reference) in projects: aosp_15_r20

Searched refs:_experimental_distribute_values_from_function (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A D	distribute_lib.py	`1753 …return self._extended._experimental_distribute_values_from_function( # pylint: disable=protected-… 2293 def _experimental_distribute_values_from_function(self, value_fn): member in StrategyExtendedV2 3660 def _experimental_distribute_values_from_function(self, value_fn): member in _DefaultDistributionExtended`
H A D	one_device_strategy.py	`341 def _experimental_distribute_values_from_function(self, value_fn): member in OneDeviceExtended`
H A D	parameter_server_strategy.py	`400 def _experimental_distribute_values_from_function(self, value_fn): member in ParameterServerStrategyExtended`
H A D	mirrored_strategy.py	`597 def _experimental_distribute_values_from_function(self, value_fn): member in MirroredExtended`
H A D	collective_all_reduce_strategy.py	`683 def _experimental_distribute_values_from_function(self, value_fn): member in CollectiveAllReduceExtended`
H A D	tpu_strategy.py	`1037 def _experimental_distribute_values_from_function(self, value_fn): member in TPUExtended`