| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local_test.cc | 72 TF_ASSERT_OK(device_mgr_->LookupDevice(kTaskName + "/device:CPU:0", &cpu0)); in TEST_F()
115 TF_ASSERT_OK(device_mgr_->LookupDevice(kTaskName + "/device:CPU:2", &cpu2)); in TEST_F()
134 TF_ASSERT_OK(device_mgr_->LookupDevice(kTaskName + "/device:CPU:1", &cpu1)); in TEST_F()
172 TF_ASSERT_OK(device_mgr_->LookupDevice(kTaskName + "/device:CPU:0", &cpu0)); in TEST_F()
194 TF_ASSERT_OK(device_mgr_->LookupDevice(kTaskName + "/device:CPU:0", &cpu0)); in TEST_F()
216 TF_ASSERT_OK(device_mgr_->LookupDevice(kTaskName + "/device:CPU:0", &cpu0)); in TEST_F()
237 TF_ASSERT_OK(device_mgr_->LookupDevice(kTaskName + "/device:CPU:0", &cpu0)); in TEST_F()
|
| D | device_mgr.h | 58 virtual Status LookupDevice(StringPiece name, Device** device) const = 0;
94 Status LookupDevice(StringPiece name, Device** device) const override;
130 Status LookupDevice(StringPiece name, Device** device) const override;
|
| D | device_resolver_local.cc | 26 Status s = dev_mgr_->LookupDevice(device, &dev); in GetDeviceAttributes()
|
| D | rendezvous_mgr.cc | 65 Status s = device_mgr->LookupDevice(parsed.src_device, &src_device); in SameWorkerRecvDone()
71 s = device_mgr->LookupDevice(parsed.dst_device, &dst_device); in SameWorkerRecvDone()
|
| D | dynamic_device_mgr.cc | 79 Status DynamicDeviceMgr::LookupDevice(StringPiece name, Device** device) const { in LookupDevice() function in tensorflow::DynamicDeviceMgr
192 TF_RETURN_IF_ERROR(LookupDevice(name, &device)); in RemoveDevicesByName()
|
| D | collective_util.cc | 40 Status status = dev_mgr->LookupDevice(device_name, device); in InitializeDeviceAndLocality()
|
| D | device_mgr.cc | 112 Status StaticDeviceMgr::LookupDevice(StringPiece name, Device** device) const { in LookupDevice() function in tensorflow::StaticDeviceMgr
|
| D | collective_rma_local.cc | 42 Status status = dev_mgr_->LookupDevice(peer_device, &from_device); in RecvFromPeer()
|
| D | buf_rendezvous.cc | 156 Status consumebuf_status = dev_mgr_->LookupDevice(device_name, &device); in ConsumeBuf()
|
| D | graph_execution_state.cc | 351 Status LookupDevice(const DeviceSet& device_set, const string& tensor_name, in LookupDevice() function
514 TF_RETURN_IF_ERROR(LookupDevice(*device_set_, feed, in PruneGraph()
534 TF_RETURN_IF_ERROR(LookupDevice(*device_set_, fetch, in PruneGraph()
|
| D | process_function_library_runtime_test.cc | 114 TF_CHECK_OK(device_mgr_->LookupDevice( in ProcessFunctionLibraryRuntimeTest()
116 TF_CHECK_OK(device_mgr_->LookupDevice( in ProcessFunctionLibraryRuntimeTest()
122 ->LookupDevice("/job:a/replica:0/task:0/device:CPU:2", &device2_ptr) in ProcessFunctionLibraryRuntimeTest()
125 Status status = device_mgr_->LookupDevice( in ProcessFunctionLibraryRuntimeTest()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed_test.cc | 319 TF_EXPECT_OK(device_mgrs_[0]->LookupDevice(dev_name, &dst_device)); in TEST_F()
349 TF_EXPECT_OK(device_mgrs_[0]->LookupDevice(dev_name, &dst_device)); in TEST_F()
384 TF_EXPECT_OK(device_mgrs_[0]->LookupDevice(dev_name, &dst_device)); in TEST_F()
412 TF_EXPECT_OK(device_mgrs_[0]->LookupDevice(dev_name, &dst_device)); in TEST_F()
|
| D | session_mgr.cc | 141 return device_mgr->LookupDevice(name, device); in CreateSession()
226 if (!local_device_mgr->LookupDevice(da.name(), &device).ok() && in UpdateSession()
227 !remote_device_mgr->LookupDevice(da.name(), &device).ok()) { in UpdateSession()
|
| D | base_rendezvous_mgr.cc | 250 Status s = sess->device_mgr()->LookupDevice(parsed.src_device, &src_device); in SameWorkerRecvDone()
256 s = sess->device_mgr()->LookupDevice(parsed.dst_device, &dst_device); in SameWorkerRecvDone()
|
| D | collective_rma_distributed.cc | 127 Status status = dev_mgr_->LookupDevice("CPU:0", &cpu_dev); in RecvFromPeer()
|
| D | graph_mgr.cc | 175 Status s = device_mgr_->LookupDevice(name, &device); in InitItem()
221 Status s = device_mgr_->LookupDevice(device_name, &unit->device); in InitItem()
|
| D | session_mgr_test.cc | 101 session->remote_device_mgr()->LookupDevice(local_cpu->name(), &device)); in TEST_F()
|
| D | collective_param_resolver_distributed_test.cc | 226 TF_CHECK_OK(device_mgrs_[task_name]->LookupDevice(device_name, &device)); in IssueRequest()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_mgr.cc | 157 if (parent_->local_device_mgr()->LookupDevice(in.op_device(), &device).ok() || in DeserializeRemoteTensorHandle()
158 parent_->local_device_mgr()->LookupDevice(in.device(), &device).ok()) { in DeserializeRemoteTensorHandle()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | py_client.cc | 80 pjrt_client_->LookupDevice(device_id)); in GetDefaultDeviceAssignment()
96 pjrt_client_->LookupDevice(device_id)); in GetDefaultDeviceAssignment1D()
111 pjrt_client_->LookupDevice(device->id())); in PjRtBufferFromPyval()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lift_variables.cc | 106 if (!(device_manager->LookupDevice(resource_handle.device(), &device) in LiftVariablesFromSession()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr.cc | 261 s = sess->device_mgr()->LookupDevice(parsed.dst_device, &dst_device); in RecvFromRemoteAsync()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_client.h | 163 virtual StatusOr<PjRtDevice*> LookupDevice(int device_id) const = 0;
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.cc | 501 static std::shared_ptr<PjRtDevice> LookupDevice(const PyTpuClient& client, in LookupDevice() function
524 std::shared_ptr<PjRtDevice> device = LookupDevice(*client_, device_id); in PyTpuExecutable()
549 std::shared_ptr<PjRtDevice> device = LookupDevice(*client_, device_id); in ExecuteHelper()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | partitioned_function_ops.cc | 211 TF_RETURN_IF_ERROR(lib->device_mgr()->LookupDevice("CPU:0", &cpu_device)); in Instantiate()
|