Searched refs:_local_results (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib.py | 1487 return self._extended._local_results(value) # pylint: disable=protected-access
1506 return self._extended._local_results(value) # pylint: disable=protected-access
1803 return self._extended._local_results(
2222 return self._local_results(self.reduce_to(reduce_op, value, dst))[0]
2543 def _local_results(self, distributed_value): member in StrategyExtendedV2
2563 value = nest.flatten(self._local_results(value))
3487 return nest.map_structure(self._local_results, result)
3492 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 | 564 return nest.map_structure(self._local_results, result)
574 return nest.map_structure(self._local_results, result)
576 def _local_results(self, val): member in ParameterServerStrategyExtended
|
| D | distribute_utils.py | 168 … return nest.map_structure(extended._local_results, regrouped) # pylint: disable=protected-access
|
| D | mirrored_strategy.py | 613 ctx.last_step_outputs[name] = self._local_results(output)
796 def _local_results(self, val): member in MirroredExtended
|
| D | tpu_strategy.py | 1305 def _local_results(self, val): member in TPUExtended
1397 return nest.map_structure(self._local_results, result)
|
| D | distribute_lib_test.py | 94 def _local_results(self, value): member in _TestExtended
|