Searched defs:colocate_with (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | numpy_dataset.py | 76 def one_host_numpy_dataset(numpy_input, colocate_with, session): argument
|
| D | one_device_strategy.py | 365 def _update_non_slot(self, colocate_with, fn, args, kwargs, group): argument
|
| D | parameter_server_strategy.py | 521 def _update_non_slot(self, colocate_with, fn, args, kwargs, group): argument
|
| D | distribute_lib.py | 1608 self, colocate_with, fn, args=(), kwargs=None, group=True): argument
1630 def _update_non_slot(self, colocate_with, fn, args, kwargs, group): argument
2245 def _update_non_slot(self, colocate_with, fn, args, kwargs, should_group): argument
|
| D | distribute_lib_test.py | 119 def _update_non_slot(self, colocate_with, fn, args, kwargs, group): argument
|
| D | tpu_strategy.py | 715 def _update_non_slot(self, colocate_with, fn, args, kwargs, group): argument
|
| D | mirrored_strategy.py | 825 def _update_non_slot(self, colocate_with, fn, args, kwargs, group): argument
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 813 def _create_non_slot_variable(self, initial_value, name, colocate_with): argument
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops.py | 4137 def colocate_with(self, op, ignore_existing=False): member in Graph
5139 def colocate_with(op, ignore_existing=False): function
|