Searched refs:select_device_mirrored (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | mirrored_strategy.py | 722 *values.select_device_mirrored(d, args),
723 **values.select_device_mirrored(d, kwargs)))
733 updates.append(fn(*values.select_device_mirrored(d, args),
734 **values.select_device_mirrored(d, kwargs)))
|
| D | tpu_strategy.py | 495 *values.select_device_mirrored(d, args),
496 **values.select_device_mirrored(d, kwargs)))
|
| D | values.py | 1393 def select_device_mirrored(device, structured): function
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | values_test.py | 230 values.select_device_mirrored(_device_str(0), result)
232 values.select_device_mirrored(_device_str(1), result)
263 values.select_device_mirrored(_device_str(0), result))
265 values.select_device_mirrored(_device_str(1), result))
|