Home
last modified time | relevance | path

Searched defs:LookupDevice (Results 1 – 5 of 5) 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.cc79 Status DynamicDeviceMgr::LookupDevice(StringPiece name, Device** device) const { in LookupDevice() function in tensorflow::DynamicDeviceMgr
Dgraph_execution_state.cc351 Status LookupDevice(const DeviceSet& device_set, const string& tensor_name, in LookupDevice() function
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.h144 StatusOr<PjRtDevice*> LookupDevice(int device_id) const override { in LookupDevice() function
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc501 static std::shared_ptr<PjRtDevice> LookupDevice(const PyTpuClient& client, in LookupDevice() function