Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dcollective_all_reduce_strategy.py683 def _experimental_distribute_values_from_function(self, value_fn): member in CollectiveAllReduceExtended
H A Done_device_strategy.py341 def _experimental_distribute_values_from_function(self, value_fn): member in OneDeviceExtended
H A Dparameter_server_strategy.py400 def _experimental_distribute_values_from_function(self, value_fn): member in ParameterServerStrategyExtended
H A Dmirrored_strategy.py597 def _experimental_distribute_values_from_function(self, value_fn): member in MirroredExtended
H A Ddistribute_lib.py2293 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 Dtpu_strategy.py1037 def _experimental_distribute_values_from_function(self, value_fn): member in TPUExtended