| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib.py | 920 def colocate_vars_with(self, colocate_with_variable): argument
922 return self._extended.colocate_vars_with(colocate_with_variable)
2153 def colocate_vars_with(self, colocate_with_variable): argument
2193 kwargs["colocate_with"] = colocate_with_variable
2197 self._validate_colocate_with_variable(colocate_with_variable)
2200 def _validate_colocate_with_variable(self, colocate_with_variable): argument
3417 def colocate_vars_with(self, colocate_with_variable): argument
3420 return ops.colocate_with(colocate_with_variable)
|
| D | one_device_strategy.py | 286 def _validate_colocate_with_variable(self, colocate_with_variable): argument
287 distribute_utils.validate_colocate(colocate_with_variable, self)
|
| D | parameter_server_strategy.py | 338 def _validate_colocate_with_variable(self, colocate_with_variable): argument
339 distribute_utils.validate_colocate(colocate_with_variable, self)
|
| D | mirrored_strategy.py | 533 def _validate_colocate_with_variable(self, colocate_with_variable): argument
535 colocate_with_variable, self)
|
| D | tpu_strategy.py | 874 def _validate_colocate_with_variable(self, colocate_with_variable): argument
875 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.-mirrored-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.-multi-worker-mirrored-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.-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.-strategy.pbtxt | 24 … 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.-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.-strategy.pbtxt | 24 … 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.-mirrored-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.-one-device-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"
|
| D | tensorflow.distribute.experimental.-central-storage-strategy.pbtxt | 25 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|