| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib.py | 963 def colocate_vars_with(self, colocate_with_variable): argument
965 return self._extended.colocate_vars_with(colocate_with_variable)
2230 def colocate_vars_with(self, colocate_with_variable): argument
2270 kwargs["colocate_with"] = colocate_with_variable
2274 self._validate_colocate_with_variable(colocate_with_variable)
2277 def _validate_colocate_with_variable(self, colocate_with_variable): argument
3646 def colocate_vars_with(self, colocate_with_variable): argument
3649 return ops.colocate_with(colocate_with_variable)
|
| D | one_device_strategy.py | 284 def _validate_colocate_with_variable(self, colocate_with_variable): argument
285 distribute_utils.validate_colocate(colocate_with_variable, self)
|
| D | parameter_server_strategy.py | 336 def _validate_colocate_with_variable(self, colocate_with_variable): argument
337 distribute_utils.validate_colocate(colocate_with_variable, self)
|
| D | mirrored_strategy.py | 538 def _validate_colocate_with_variable(self, colocate_with_variable): argument
540 colocate_with_variable, self)
|
| D | tpu_strategy.py | 931 def _validate_colocate_with_variable(self, colocate_with_variable): argument
932 distribute_utils. validate_colocate(colocate_with_variable, self)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 27 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-multi-worker-mirrored-strategy.pbtxt | 25 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 25 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt | 25 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.experimental.-central-storage-strategy.pbtxt | 25 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-one-device-strategy.pbtxt | 25 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-parameter-server-strategy.pbtxt | 25 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 26 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-strategy.pbtxt | 24 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 25 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-t-p-u-strategy.pbtxt | 25 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 52 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 25 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-strategy.pbtxt | 24 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt | 25 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.experimental.-central-storage-strategy.pbtxt | 25 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-one-device-strategy.pbtxt | 25 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 25 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 29 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|