Home
last modified time | relevance | path

Searched defs:worker_device (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Ddevice_setter.py72 def __init__(self, ps_tasks, ps_device, worker_device, merge_devices, ps_ops, argument
133 worker_device="/job:worker", argument
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dcollective_all_reduce_strategy.py360 def _initialize_local_devices(self, cluster_resolver, worker_device): argument
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dengine.cpp107 static thread_local int worker_device = NO_DEVICE; variable