| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | dynamic_device_mgr_test.cc | 71 EXPECT_EQ(dm->ListDevices().size(), 0); in TEST()
77 EXPECT_EQ(dm->ListDevices().size(), 2); in TEST()
91 EXPECT_EQ(dm->ListDevices().size(), 2); in TEST()
95 EXPECT_EQ(dm->ListDevices().size(), 1); in TEST()
140 EXPECT_EQ(dm->ListDevices().size(), 2); in TEST()
144 EXPECT_EQ(dm->ListDevices().size(), 1); in TEST()
155 EXPECT_EQ(dm->ListDevices().size(), 1); in TEST()
174 EXPECT_EQ(dm->ListDevices().size(), 1); in TEST()
179 EXPECT_EQ(dm->ListDevices().size(), 1); // d0 *not* removed. in TEST()
191 EXPECT_EQ(dm->ListDevices().size(), 1); in TEST()
[all …]
|
| D | device_mgr.h | 48 virtual std::vector<Device*> ListDevices() const = 0;
91 std::vector<Device*> ListDevices() const override;
131 std::vector<Device*> ListDevices() const override;
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle_test.cc | 120 std::vector<Device*> ListDevices() const { in ListDevices() function in tensorflow::PackedTensorHandleTest
121 return device_mgr_->ListDevices(); in ListDevices()
147 Device* d0 = ListDevices().at(0); in TEST_F()
153 Device* d1 = ListDevices().at(1); in TEST_F()
161 Device* d2 = ListDevices().at(2); in TEST_F()
165 Device* d3 = ListDevices().at(3); in TEST_F()
203 EXPECT_EQ(h->device(), ListDevices().at(i)); in TEST_F()
208 TF_ASSERT_OK(h2->SetRemoteShape(shape, ListDevices().at(2), in TEST_F()
211 TF_ASSERT_OK(h3->SetRemoteShape(shape, ListDevices().at(3), in TEST_F()
223 Device* d = ListDevices().at(0); in TEST_F()
[all …]
|
| D | execute_node_test.cc | 58 Device* device0 = device_mgr.ListDevices().at(0); in TEST()
64 Device* device1 = remote_device_mgr->ListDevices().at(0); in TEST()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | session_mgr_test.cc | 54 env_.local_devices = device_mgr_->ListDevices(); in SessionMgrTest()
131 std::vector<Device*> devices_1 = session_1->device_mgr()->ListDevices(); in TEST_F()
137 std::vector<Device*> devices_2 = session_2->device_mgr()->ListDevices(); in TEST_F()
143 std::vector<Device*> devices_3 = session_3->device_mgr()->ListDevices(); in TEST_F()
149 std::vector<Device*> devices_4 = session_4->device_mgr()->ListDevices(); in TEST_F()
|
| D | local_master.cc | 134 Status LocalMaster::ListDevices(CallOptions* call_options, in ListDevices() function in tensorflow::LocalMaster
139 master_impl_->ListDevices(request, response, [&n, &ret](const Status& s) { in ListDevices()
|
| D | device_resolver_distributed.cc | 25 for (Device* device : dev_mgr->ListDevices()) { in DeviceResolverDistributed()
|
| D | master.h | 59 void ListDevices(const ListDevicesRequest* req, ListDevicesResponse* resp,
|
| D | local_master.h | 66 Status ListDevices(CallOptions* call_options,
|
| D | master_interface.h | 88 virtual Status ListDevices(CallOptions* call_options,
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine.cc | 84 TF_RETURN_IF_ERROR(session_->ListDevices(&devices)); in Provision()
225 std::vector<Device*> devices = device_mgr->ListDevices(); in GetPeakMemoryUsage()
374 for (auto d : device_mgr->ListDevices()) { in ResetSession()
454 std::vector<Device*> devices = device_mgr->ListDevices(); in ClearAllocatorStats()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_compat_request_state.cc | 91 for (auto* device : device_manager_->ListDevices()) { in KernelFallbackCompatRequestState()
118 device_manager->ListDevices()) { in KernelFallbackCompatRequestState()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_master_service.cc | 112 ENQUEUE_REQUEST(ListDevices, false); in HandleRPCsLoop()
230 master_impl_->ListDevices(&call->request, &call->response, in ListDevicesHandler()
234 ENQUEUE_REQUEST(ListDevices, false); in ListDevicesHandler()
|
| D | grpc_remote_master.cc | 82 Status ListDevices(CallOptions* call_options, in ListDevices() function in tensorflow::GrpcRemoteMaster
86 &MasterServiceStub::ListDevices); in ListDevices()
|
| D | grpc_master_service_impl.h | 60 virtual ::grpc::Status ListDevices(::grpc::ClientContext* context,
94 ::grpc::Status ListDevices(::grpc::ClientContext* context,
|
| /external/tensorflow/tensorflow/go/ |
| D | context_test.go | 41 devs, err := c.ListDevices()
|
| D | context.go | 101 func (c *Context) ListDevices() ([]Device, error) { func
|
| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.cc | 52 params.device = device_mgr->ListDevices()[0]; in CreateParams()
74 Device* device = device_mgr->ListDevices()[0]; in FromGraph()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | c_api_tfrt_distributed_impl.cc | 141 dist_context_->GetRemoteDeviceManager()->ListDevices<Device>(); in SetOrUpdateServerDef()
207 for (auto& device : tf_devices_->ListDevices()) { in PopulateRemoteDevices()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/integration_test/ |
| D | c_api_multi_client_test.cc | 129 context->ListDevices(&device_attrs); in TEST()
175 context->ListDevices(&device_attrs); in TEST()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/backends/cpu/ |
| D | cpu_registration.cc | 36 for (auto& device : device_mgr->ListDevices()) { in RegisterCpuOpHandler()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/backends/gpu/ |
| D | gpu_registration.cc | 36 for (auto& device : device_mgr->ListDevices()) { in RegisterGpuOpHandler()
|
| /external/tensorflow/tensorflow/core/tfrt/fallback/ |
| D | fallback_state.cc | 53 for (auto *d : device_manager_.ListDevices()) {
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | fake_session.h | 37 ::tensorflow::Status ListDevices(
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster.py | 67 def ListDevices(self): member in Cluster
|