Home
last modified time | relevance | path

Searched defs:LookupDevice (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Ddevice_mgr.cc112 Status StaticDeviceMgr::LookupDevice(StringPiece name, Device** device) const { in LookupDevice() function in tensorflow::StaticDeviceMgr
Ddynamic_device_mgr.cc98 Status DynamicDeviceMgr::LookupDevice(StringPiece name, Device** device) const { in LookupDevice() function in tensorflow::DynamicDeviceMgr
Dgraph_execution_state.cc352 Status LookupDevice(const DeviceSet& device_set, const string& tensor_name, in LookupDevice() function
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.h147 StatusOr<PjRtDevice*> LookupDevice(int device_id) const override { in LookupDevice() function
Dtfrt_cpu_pjrt_client.cc166 StatusOr<PjRtDevice*> TfrtCpuClient::LookupDevice(int device_id) const { in LookupDevice() function in xla::TfrtCpuClient
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc502 static std::shared_ptr<PjRtDevice> LookupDevice(const PyTpuClient& client, in LookupDevice() function