HomeSort by relevance Sort by last modified time
    Searched refs:ps_devices (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/estimator/python/estimator/
replicate_model_fn.py 202 ps_devices = [consolidation_device]
204 ps_devices = devices
208 .format(devices, ps_devices, consolidation_device))
221 local_ps_devices=ps_devices)[0] # One device, so one spec is out.
236 local_ps_devices=ps_devices)
535 ps_devices=local_ps_devices,
581 def _local_device_setter(worker_device, ps_devices, ps_strategy):
591 '{}'.format(ps_devices[ps_strategy(op)]))
replicate_model_fn_test.py     [all...]

Completed in 71 milliseconds