| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib.py | 334 def colocate_vars_with(self, colocate_with_variable): argument
336 return self._extended.colocate_vars_with(colocate_with_variable)
935 def colocate_vars_with(self, colocate_with_variable): argument
974 kwargs["colocate_with"] = colocate_with_variable
978 self._validate_colocate_with_variable(colocate_with_variable)
981 def _validate_colocate_with_variable(self, colocate_with_variable): argument
1564 def colocate_vars_with(self, colocate_with_variable): argument
1567 return ops.colocate_with(colocate_with_variable)
|
| D | one_device_strategy.py | 73 def _validate_colocate_with_variable(self, colocate_with_variable): argument
74 values.validate_colocate(colocate_with_variable, self)
|
| D | parameter_server_strategy.py | 252 def _validate_colocate_with_variable(self, colocate_with_variable): argument
253 values.validate_colocate(colocate_with_variable, self)
|
| D | tpu_strategy.py | 254 def _validate_colocate_with_variable(self, colocate_with_variable): argument
255 values.validate_colocate_tpu_variable(colocate_with_variable, self)
|
| D | mirrored_strategy.py | 566 def _validate_colocate_with_variable(self, colocate_with_variable): argument
567 values.validate_colocate_distributed_variable(colocate_with_variable, self)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-strategy.pbtxt | 19 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt | 20 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 24 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 20 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-one-device-strategy.pbtxt | 20 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 20 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-strategy-extended.pbtxt | 51 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy.pbtxt | 19 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt | 20 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 20 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 20 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 24 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-one-device-strategy.pbtxt | 20 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.distribute.-strategy-extended.pbtxt | 51 … argspec: "args=[\'self\', \'colocate_with_variable\'], varargs=None, keywords=None, defaults=None"
|