Searched refs:_local_results (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib.py | 1542 return self._extended._local_results(value) # pylint: disable=protected-access
1562 return self._extended._local_results(value) # pylint: disable=protected-access
1860 return self._extended._local_results(
2299 return self._local_results(self.reduce_to(reduce_op, value, dst))[0]
2641 def _local_results(self, val): member in StrategyExtendedV2
2679 value = nest.flatten(self._local_results(value))
3716 return nest.map_structure(self._local_results, result)
3721 def _local_results(self, distributed_value): member in _DefaultDistributionExtended
|
| D | one_device_strategy.py | 416 return nest.map_structure(self._local_results, result)
422 def _local_results(self, value): member in OneDeviceExtended
|
| D | parameter_server_strategy.py | 556 return nest.map_structure(self._local_results, result)
566 return nest.map_structure(self._local_results, result)
|
| D | distribute_utils.py | 170 … return nest.map_structure(extended._local_results, regrouped) # pylint: disable=protected-access
|
| D | parameter_server_strategy_v2.py | 925 result = self._local_results(
|
| D | distribute_lib_test.py | 91 def _local_results(self, value): member in _TestExtended
|
| D | mirrored_strategy.py | 619 ctx.last_step_outputs[name] = self._local_results(output)
|
| D | tpu_strategy.py | 1528 return nest.map_structure(self._local_results, result)
|