Home
last modified time | relevance | path

Searched refs:ListDevices (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/python/client/
Ddevice_lib.i28 static std::vector<string> ListDevices(TF_Status* out_status) { in ListDevices() function
65 %unignore tensorflow::swig::ListDevices;
70 std::vector<string> ListDevices(TF_Status* out_status);
79 return ListDevices(status)
/external/tensorflow/tensorflow/core/distributed_runtime/
Dsession_mgr_test.cc85 std::vector<Device*> devices_1 = session_1->device_mgr->ListDevices(); in TEST_F()
90 std::vector<Device*> devices_2 = session_2->device_mgr->ListDevices(); in TEST_F()
95 std::vector<Device*> devices_3 = session_3->device_mgr->ListDevices(); in TEST_F()
100 std::vector<Device*> devices_4 = session_4->device_mgr->ListDevices(); in TEST_F()
Dlocal_master.cc132 Status LocalMaster::ListDevices(CallOptions* call_options, in ListDevices() function in tensorflow::LocalMaster
137 master_impl_->ListDevices(request, response, [&n, &ret](const Status& s) { in ListDevices()
Dmaster.h58 void ListDevices(const ListDevicesRequest* req, ListDevicesResponse* resp,
Dlocal_master.h66 Status ListDevices(CallOptions* call_options,
Dmaster_interface.h85 virtual Status ListDevices(CallOptions* call_options,
Dmaster_test.cc148 TEST_F(MasterTest, ListDevices) { in TEST_F() argument
152 const Status s = FromGrpcStatus(master_->ListDevices(&ctx, req, &resp)); in TEST_F()
Dmaster_session.h90 Status ListDevices(ListDevicesResponse* resp) const;
Dmaster.cc524 void Master::ListDevices(const ListDevicesRequest* req, in ListDevices() function in tensorflow::Master
536 Status s = session->ListDevices(resp); in ListDevices()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_master_service.cc112 ENQUEUE_REQUEST(ListDevices, false); in HandleRPCsLoop()
220 master_impl_->ListDevices(&call->request, &call->response, in ListDevicesHandler()
224 ENQUEUE_REQUEST(ListDevices, false); in ListDevicesHandler()
Dgrpc_remote_master.cc83 Status ListDevices(CallOptions* call_options, in ListDevices() function in tensorflow::GrpcRemoteMaster
88 &MasterServiceStub::ListDevices); in ListDevices()
Dgrpc_master_service_impl.h76 virtual ::grpc::Status ListDevices(::grpc::ClientContext* context,
101 ::grpc::Status ListDevices(::grpc::ClientContext* context,
Dgrpc_testlib.cc73 TF_RETURN_IF_ERROR(session->ListDevices(&ret->devices_)); in MakeTestCluster()
Dgrpc_session.h97 Status ListDevices(std::vector<DeviceAttributes>* response) override;
Dgrpc_session.cc344 Status GrpcSession::ListDevices(std::vector<DeviceAttributes>* response) { in ListDevices() function in tensorflow::GrpcSession
364 Status s = master_->ListDevices(&call_options, &req, &resp); in ListDevices()
Dgrpc_master_service_impl.cc104 ::grpc::Status MasterService::Stub::ListDevices( in ListDevices() function in tensorflow::grpc::MasterService::Stub
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine.cc82 TF_RETURN_IF_ERROR(session_->ListDevices(&devices)); in Provision()
213 std::vector<Device*> devices = device_mgr->ListDevices(); in GetPeakMemoryUsage()
402 std::vector<Device*> devices = device_mgr->ListDevices(); in ClearAllocatorStats()
/external/tensorflow/tensorflow/core/common_runtime/
Ddevice_mgr.h46 std::vector<Device*> ListDevices() const;
Dprocess_function_library_runtime.cc43 for (Device* d : device_mgr->ListDevices()) { in ProcessFunctionLibraryRuntime()
66 for (Device* d : device_mgr->ListDevices()) { in ProcessFunctionLibraryRuntime()
Ddevice_mgr.cc66 std::vector<Device*> DeviceMgr::ListDevices() const { in ListDevices() function in tensorflow::DeviceMgr
Ddirect_session.cc299 for (auto d : device_mgr_->ListDevices()) { in DirectSession()
321 for (auto d : device_mgr_->ListDevices()) { in ~DirectSession()
324 for (auto d : device_mgr_->ListDevices()) { in ~DirectSession()
1459 ::tensorflow::Status DirectSession::ListDevices( in ListDevices() function in tensorflow::DirectSession
/external/tensorflow/tensorflow/core/public/
Dsession.h180 virtual Status ListDevices(std::vector<DeviceAttributes>* response) = 0;
/external/tensorflow/tensorflow/python/grappler/
Dcluster.py73 def ListDevices(self): member in Cluster
/external/tensorflow/tensorflow/cc/profiler/
Dprofiler_test.cc128 TF_CHECK_OK(session->ListDevices(&resp)); in TEST_F()
/external/tensorflow/tensorflow/go/
Dsession_test.go293 devices, err := s.ListDevices()

12