| /external/tensorflow/tensorflow/python/distribute/ |
| D | collective_all_reduce_strategy.py | 119 local_devices = tuple("/device:GPU:%d" % i for i in range(num_gpus))
121 local_devices = ("/device:CPU:0",)
126 super(CollectiveAllReduceExtended, self)._initialize_local(local_devices)
148 local_devices)
185 local_devices = tuple("%s/device:GPU:%d" % (self._worker_device, i)
188 local_devices = (self._worker_device,)
191 super(CollectiveAllReduceExtended, self)._initialize_local(local_devices)
215 task_id, self._num_workers, local_devices,
|
| /external/libdrm/ |
| D | xf86drm.c | 3682 static void drmFoldDuplicatedDevices(drmDevicePtr local_devices[], int count) in drmFoldDuplicatedDevices() argument
3688 if (drmDevicesEqual(local_devices[i], local_devices[j])) { in drmFoldDuplicatedDevices()
3689 local_devices[i]->available_nodes |= local_devices[j]->available_nodes; in drmFoldDuplicatedDevices()
3690 node_type = log2(local_devices[j]->available_nodes); in drmFoldDuplicatedDevices()
3691 memcpy(local_devices[i]->nodes[node_type], in drmFoldDuplicatedDevices()
3692 local_devices[j]->nodes[node_type], drmGetMaxNodeName()); in drmFoldDuplicatedDevices()
3693 drmFreeDevice(&local_devices[j]); in drmFoldDuplicatedDevices()
3786 drmDevicePtr *local_devices; in drmGetDevice2()
3816 local_devices = calloc(max_count, sizeof(drmDevicePtr)); in drmGetDevice2()
3817 if (local_devices == NULL) in drmGetDevice2()
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master.cc | 72 CHECK(!env->local_devices.empty()); in Master()
142 finder.GetRemoteDevices(env->local_devices, out_remote); in GetRemoteDevices()
181 CHECK_GT(env_->local_devices.size(), 0) << "No local devices provided."; in DeviceFinder()
182 const string& local_device_name = env_->local_devices[0]->name(); in DeviceFinder()
402 if (env_->local_devices[0]->parsed_name().job == job.name() && in CreateSession()
403 env_->local_devices[0]->parsed_name().task == task.first) { in CreateSession()
452 for (Device* d : env_->local_devices) { in CreateSession()
608 for (Device* dev : env_->local_devices) { in ListDevices()
|
| D | session_mgr.cc | 74 CHECK(!worker_env_->local_devices.empty()) in CreateSession()
82 for (Device* d : worker_env_->local_devices) { in CreateSession()
|
| D | worker_env.h | 48 std::vector<Device*> local_devices; member
|
| D | master_env.h | 75 std::vector<Device*> local_devices; member
|
| D | session_mgr_test.cc | 51 env_.local_devices = device_mgr_->ListDevices(); in SessionMgrTest()
|
| D | worker.cc | 354 for (Device* d : env_->local_devices) { in CleanupGraphAsync()
|
| D | master_session.cc | 1191 VLOG(1) << "Session " << handle_ << " #local " << env->local_devices.size() in MasterSession()
1385 for (Device* dev : env_->local_devices) { in ListDevices()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer_test.cc | 98 std::vector<std::unique_ptr<Device>> local_devices; in Init() local
102 local_devices.emplace_back(std::move(gpus_[rank])); in Init()
105 int num_gpus = local_devices.size(); in Init()
106 for (const auto& device : local_devices) { in Init()
110 if (!dev_mgr_) dev_mgr_.reset(new DeviceMgr(std::move(local_devices))); in Init()
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | session_debug_multi_gpu_test.py | 50 local_devices = device_lib.list_local_devices()
52 for device in local_devices:
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_gatherer_test.cc | 139 std::vector<std::unique_ptr<Device>> local_devices; in Init() local
149 local_devices.push_back(absl::make_unique<ThreadPoolDevice>( in Init()
157 local_devices.push_back(std::move(gpu_devices_[dev_idx])); in Init()
165 LOG(ERROR) << "resetting dev_mgr for " << local_devices.size() in Init()
167 dev_mgr_.reset(new DeviceMgr(std::move(local_devices))); in Init()
|
| D | ring_reducer_test.cc | 161 std::vector<std::unique_ptr<Device>> local_devices; in Init() local
171 local_devices.push_back(absl::make_unique<ThreadPoolDevice>( in Init()
179 local_devices.push_back(std::move(gpu_devices_[dev_idx])); in Init()
187 LOG(ERROR) << "resetting dev_mgr for " << local_devices.size() in Init()
189 dev_mgr_.reset(new DeviceMgr(std::move(local_devices))); in Init()
|
| D | hierarchical_tree_broadcaster_test.cc | 221 std::vector<std::unique_ptr<Device>> local_devices; in Init() local
231 local_devices.push_back(absl::make_unique<ThreadPoolDevice>( in Init()
239 local_devices.push_back(std::move(gpu_devices_[dev_idx])); in Init()
247 dev_mgr_.reset(new DeviceMgr(std::move(local_devices))); in Init()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_server_lib.cc | 161 master_env_.local_devices = worker_env_.device_mgr->ListDevices(); in Init()
162 worker_env_.local_devices = worker_env_.device_mgr->ListDevices(); in Init()
168 if (!DeviceNameUtils::SplitDeviceName(master_env_.local_devices[0]->name(), in Init()
|
| /external/tensorflow/tensorflow/contrib/mpi/ |
| D | mpi_rendezvous_mgr.cc | 48 auto parsed = env->local_devices[0]->parsed_name(); in MPIRendezvousMgr()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 73 worker_env_.local_devices = device_mgr_->ListDevices();
|