| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device_resolver_local.cc | 27 Status s = dev_mgr_->LookupDevice(device_name, &dev); in GetDeviceLocalitiesAsync()
42 Status s = dev_mgr_->LookupDevice(device, &dev); in GetLocalityAsync()
|
| D | collective_rma_local_test.cc | 65 TF_ASSERT_OK(device_mgr_->LookupDevice(kTaskName + "/device:CPU:0", &cpu0)); in TEST_F()
108 TF_ASSERT_OK(device_mgr_->LookupDevice(kTaskName + "/device:CPU:2", &cpu2)); in TEST_F()
127 TF_ASSERT_OK(device_mgr_->LookupDevice(kTaskName + "/device:CPU:1", &cpu1)); in TEST_F()
|
| D | rendezvous_mgr.cc | 88 Status s = device_mgr_->LookupDevice(parsed.src_device, &src_device); in SameWorkerRecvDone()
94 s = device_mgr_->LookupDevice(parsed.dst_device, &dst_device); in SameWorkerRecvDone()
|
| D | device_mgr.h | 59 Status LookupDevice(StringPiece name, Device** device) const;
|
| D | collective_util.cc | 40 Status status = dev_mgr->LookupDevice(device_name, device); in InitializeDeviceAndLocality()
|
| D | device_mgr.cc | 97 Status DeviceMgr::LookupDevice(StringPiece name, Device** device) const { in LookupDevice() function in tensorflow::DeviceMgr
|
| D | process_function_library_runtime_test.cc | 76 TF_CHECK_OK(device_mgr_->LookupDevice( in Init()
78 TF_CHECK_OK(device_mgr_->LookupDevice( in Init()
81 Status status = device_mgr_->LookupDevice( in Init()
|
| D | graph_execution_state.cc | 292 Status LookupDevice(const DeviceSet& device_set, const string& tensor_name, in LookupDevice() function
455 TF_RETURN_IF_ERROR(LookupDevice(*device_set_, feed, in PruneGraph()
475 TF_RETURN_IF_ERROR(LookupDevice(*device_set_, fetch, in PruneGraph()
|
| D | process_function_library_runtime.cc | 185 if (!device_mgr_->LookupDevice(device_name, &device).ok()) { in GetFLR()
623 TF_RETURN_IF_ERROR(device_mgr_->LookupDevice("CPU:0", &cpu_device)); in InstantiateMultiDevice()
|
| D | ring_gatherer_test.cc | 384 TF_CHECK_OK(parent_->dev_mgr_->LookupDevice(dev_name, &device_)) in DeviceInstance()
|
| D | direct_session.cc | 1249 TF_RETURN_IF_ERROR(device_mgr_->LookupDevice(partition_name, &device)); in CreateExecutors()
1600 s = device_mgr_->LookupDevice(partition_name, &d); in CreateGraphs()
|
| /external/tensorflow/tensorflow/contrib/mpi/ |
| D | mpi_rendezvous_mgr.cc | 77 s = env_->device_mgr->LookupDevice(parsed.dst_device, &dst_device); in RecvFromRemoteAsync()
110 s = env_->device_mgr->LookupDevice(parsed.dst_device, &dst_device); in RecvFromRemoteAsync()
217 Status s = this->worker_env_2->device_mgr->LookupDevice( in AddRequest()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed_test.cc | 275 TF_EXPECT_OK(device_mgrs_[0]->LookupDevice(dev_name, &dst_device)); in TEST_F()
303 TF_EXPECT_OK(device_mgrs_[0]->LookupDevice(dev_name, &dst_device)); in TEST_F()
335 TF_EXPECT_OK(device_mgrs_[0]->LookupDevice(dev_name, &dst_device)); in TEST_F()
|
| D | device_resolver_distributed.cc | 33 Status s = dev_mgr_->LookupDevice(device, &dev); in GetLocalityAsync()
|
| D | base_rendezvous_mgr.cc | 256 Status s = sess->device_mgr()->LookupDevice(parsed.src_device, &src_device); in SameWorkerRecvDone()
262 s = sess->device_mgr()->LookupDevice(parsed.dst_device, &dst_device); in SameWorkerRecvDone()
|
| D | collective_rma_distributed.cc | 123 Status status = dev_mgr_->LookupDevice("CPU:0", &cpu_dev); in RecvFromPeer()
|
| D | graph_mgr.cc | 163 Status s = device_mgr_->LookupDevice(name, &device); in InitItem()
209 Status s = device_mgr_->LookupDevice(device_name, &unit->device); in InitItem()
|
| D | worker.cc | 435 TF_RETURN_IF_ERROR(env_->device_mgr->LookupDevice(local_name, src_dev)); in PrepareRecvTensor()
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma_rendezvous_mgr.cc | 70 s = env_->device_mgr->LookupDevice(parsed.dst_device, &dst_dev); in RecvFromRemoteAsync()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_rendezvous_mgr.cc | 153 Status s = sess->device_mgr()->LookupDevice(parsed.dst_device, &dst_device); in RecvFromRemoteAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr.cc | 242 s = sess->device_mgr()->LookupDevice(parsed.dst_device, &dst_device); in RecvFromRemoteAsync()
|
| D | grpc_worker_service.cc | 588 s = env_->device_mgr->LookupDevice("CPU:0", &cpu_dev); in RecvBufAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 248 return ctx->local_device_mgr()->LookupDevice(d->name(), &tmp).ok(); in IsLocal()
282 auto status = ctx->local_device_mgr()->LookupDevice(device_name, device); in FindDeviceFromName()
288 return ctx->remote_device_mgr()->LookupDevice(device_name, device); in FindDeviceFromName()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | partitioned_function_ops.cc | 186 TF_RETURN_IF_ERROR(lib->device_mgr()->LookupDevice("CPU:0", &cpu_device)); in Instantiate()
|
| D | collective_nccl_reducer_test.cc | 219 TF_CHECK_OK(parent_->dev_mgr_->LookupDevice(device_name_, &device_)) in DeviceInstance()
|