| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| worker_env.h | 47 // REQUIRES: !local_devices.empty().
48 std::vector<Device*> local_devices; member in struct:tensorflow::WorkerEnv
|
| master_env.h | 74 // REQUIRES: !local_devices.empty().
75 std::vector<Device*> local_devices; member in struct:tensorflow::MasterEnv
|
| master.cc | 72 CHECK(!env->local_devices.empty());
142 finder.GetRemoteDevices(env->local_devices, out_remote);
181 CHECK_GT(env_->local_devices.size(), 0) << "No local devices provided.";
182 const string& local_device_name = env_->local_devices[0]->name();
402 if (env_->local_devices[0]->parsed_name().job == job.name() &&
403 env_->local_devices[0]->parsed_name().task == task.first) {
452 for (Device* d : env_->local_devices) {
608 for (Device* dev : env_->local_devices) {
|
| session_mgr.cc | 74 CHECK(!worker_env_->local_devices.empty())
75 << "The WorkerEnv must have at least one device in `local_devices`.";
82 for (Device* d : worker_env_->local_devices) {
|
| session_mgr_test.cc | 51 env_.local_devices = device_mgr_->ListDevices();
|
| worker.cc | 354 for (Device* d : env_->local_devices) {
|
| master_session.cc | [all...] |
| /external/tensorflow/tensorflow/python/debug/lib/ |
| session_debug_multi_gpu_test.py | 50 local_devices = device_lib.list_local_devices()
52 for device in local_devices:
|
| /external/tensorflow/tensorflow/python/distribute/ |
| 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)
147 logging.info("CollectiveAllReduceStrategy with local_devices = %r",
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)
213 "task_type = %r, task_id = %r, num_workers = %r, local_devices = %r, "
215 task_id, self._num_workers, local_devices,
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| collective_nccl_reducer_test.cc | 98 std::vector<std::unique_ptr<Device>> local_devices; local
102 local_devices.emplace_back(std::move(gpus_[rank]));
105 int num_gpus = local_devices.size();
106 for (const auto& device : local_devices) {
110 if (!dev_mgr_) dev_mgr_.reset(new DeviceMgr(std::move(local_devices)));
|
| /external/libdrm/ |
| xf86drm.c | 3952 drmDevicePtr *local_devices; local
[all...] |
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| grpc_server_lib.cc | 161 master_env_.local_devices = worker_env_.device_mgr->ListDevices();
162 worker_env_.local_devices = worker_env_.device_mgr->ListDevices();
168 if (!DeviceNameUtils::SplitDeviceName(master_env_.local_devices[0]->name(),
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| ring_gatherer_test.cc | 139 std::vector<std::unique_ptr<Device>> local_devices; local
149 local_devices.push_back(absl::make_unique<ThreadPoolDevice>(
157 local_devices.push_back(std::move(gpu_devices_[dev_idx]));
165 LOG(ERROR) << "resetting dev_mgr for " << local_devices.size()
167 dev_mgr_.reset(new DeviceMgr(std::move(local_devices)));
|
| ring_reducer_test.cc | 161 std::vector<std::unique_ptr<Device>> local_devices; local
171 local_devices.push_back(absl::make_unique<ThreadPoolDevice>(
179 local_devices.push_back(std::move(gpu_devices_[dev_idx]));
187 LOG(ERROR) << "resetting dev_mgr for " << local_devices.size()
189 dev_mgr_.reset(new DeviceMgr(std::move(local_devices)));
|
| hierarchical_tree_broadcaster_test.cc | 221 std::vector<std::unique_ptr<Device>> local_devices; local
231 local_devices.push_back(absl::make_unique<ThreadPoolDevice>(
239 local_devices.push_back(std::move(gpu_devices_[dev_idx]));
247 dev_mgr_.reset(new DeviceMgr(std::move(local_devices)));
[all...] |
| /external/tensorflow/tensorflow/contrib/mpi/ |
| mpi_rendezvous_mgr.cc | 48 auto parsed = env->local_devices[0]->parsed_name();
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| eager_service_impl_test.cc | 73 worker_env_.local_devices = device_mgr_->ListDevices();
|
