Searched refs:device_idx (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | outfeed_receiver.cc | 185 void DeviceListenerThreadLoop(int device_idx);
188 Status SendShutdownOutfeedHeader(int device_idx);
255 for (int device_idx = 0; device_idx < devices_.size(); ++device_idx) { in Start() local
257 [this, device_idx]() { DeviceListenerThreadLoop(device_idx); }); in Start()
268 for (int device_idx = 0; device_idx < devices_.size(); ++device_idx) { in Shutdown() local
269 CHECK(SendShutdownOutfeedHeader(device_idx).ok()); in Shutdown()
282 void OutfeedReceiverImpl::DeviceListenerThreadLoop(int device_idx) { in DeviceListenerThreadLoop() argument
287 PjRtDevice* device = devices_[device_idx]; in DeviceListenerThreadLoop()
387 Status OutfeedReceiverImpl::SendShutdownOutfeedHeader(int device_idx) { in SendShutdownOutfeedHeader() argument
388 const PjRtDevice* device = devices_[device_idx]; in SendShutdownOutfeedHeader()
[all …]
|
| /external/autotest/client/common_lib/cros/ |
| D | power_cycle_usb_util.py | 268 device_idx =[]
278 device_idx.append( target_device_idx)
279 return (bus_idx, device_idx)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops_test.py | 75 for device_idx, device in enumerate(devices):
77 v = value(device_idx)
79 v = value[device_idx]
242 value_fn = lambda device_idx: inputs[pid * len(devices) + device_idx] argument
285 def value_fn(device_idx, idx=i): argument
286 return inputs[pid * len(devices) + device_idx][idx]
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager_test.cc | 427 const int device_idx = GlobalRank(num_ranks_per_node, node, local_rank); in GetDevice() local
428 CHECK_LT(device_idx, devices_->size()); in GetDevice()
429 return (*devices_)[device_idx].get(); in GetDevice()
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 545 for (int device_idx = 0; device_idx < devices.size(); device_idx++) { in PYBIND11_MODULE() local
546 tensorflow::Device* device = devices[device_idx]; in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_test.py | 365 for device_idx in range(group_size):
366 with ops.device('/{}:{}'.format(device, device_idx)):
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_param_resolver_local_test.cc | 211 void InitializeCollectiveParamsForBroadcast(int instance_key, int device_idx, in InitializeCollectiveParamsForBroadcast() argument
|