Searched refs:_validate_colocate_with_variable (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ | ||
| H A D | one_device_strategy.py | 284 def _validate_colocate_with_variable(self, colocate_with_variable): member in OneDeviceExtended |
| H A D | distribute_lib.py | 2274 self._validate_colocate_with_variable(colocate_with_variable) |
| H A D | parameter_server_strategy.py | 336 def _validate_colocate_with_variable(self, colocate_with_variable): member in ParameterServerStrategyExtended |
| H A D | mirrored_strategy.py | 538 def _validate_colocate_with_variable(self, colocate_with_variable): member in MirroredExtended |
| H A D | tpu_strategy.py | 931 def _validate_colocate_with_variable(self, colocate_with_variable): member in TPUExtended |