| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | dynamic_device_mgr_test.cc | 47 EXPECT_EQ(dm->ListDevices().size(), 0); in TEST()
53 EXPECT_EQ(dm->ListDevices().size(), 2); in TEST()
66 EXPECT_EQ(dm->ListDevices().size(), 2); in TEST()
70 EXPECT_EQ(dm->ListDevices().size(), 1); in TEST()
83 EXPECT_EQ(dm->ListDevices().size(), 2); in TEST()
87 EXPECT_EQ(dm->ListDevices().size(), 1); in TEST()
98 EXPECT_EQ(dm->ListDevices().size(), 1); in TEST()
117 EXPECT_EQ(dm->ListDevices().size(), 1); in TEST()
122 EXPECT_EQ(dm->ListDevices().size(), 1); // d0 *not* removed. in TEST()
134 EXPECT_EQ(dm->ListDevices().size(), 1); in TEST()
[all …]
|
| D | device_mgr.h | 47 virtual std::vector<Device*> ListDevices() const = 0;
85 std::vector<Device*> ListDevices() const override;
116 std::vector<Device*> ListDevices() const override;
|
| D | collective_util.cc | 46 for (auto d : dev_mgr->ListDevices()) { in InitializeDeviceAndLocality()
|
| D | device_mgr.cc | 84 std::vector<Device*> StaticDeviceMgr::ListDevices() const { in ListDevices() function in tensorflow::StaticDeviceMgr
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | session_mgr_test.cc | 51 env_.local_devices = device_mgr_->ListDevices(); in SessionMgrTest()
128 std::vector<Device*> devices_1 = session_1->device_mgr()->ListDevices(); in TEST_F()
134 std::vector<Device*> devices_2 = session_2->device_mgr()->ListDevices(); in TEST_F()
140 std::vector<Device*> devices_3 = session_3->device_mgr()->ListDevices(); in TEST_F()
146 std::vector<Device*> devices_4 = session_4->device_mgr()->ListDevices(); in TEST_F()
|
| D | local_master.cc | 132 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()
|
| 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()
223 std::vector<Device*> devices = device_mgr->ListDevices(); in GetPeakMemoryUsage()
372 for (auto d : device_mgr->ListDevices()) { in ResetSession()
452 std::vector<Device*> devices = device_mgr->ListDevices(); in ClearAllocatorStats()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_master_service.cc | 112 ENQUEUE_REQUEST(ListDevices, false); in HandleRPCsLoop()
228 master_impl_->ListDevices(&call->request, &call->response, in ListDevicesHandler()
232 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,
|
| D | grpc_testlib.cc | 83 TF_RETURN_IF_ERROR(session->ListDevices(&ret->devices_)); in MakeTestCluster()
|
| D | grpc_session.h | 100 Status ListDevices(std::vector<DeviceAttributes>* response) override;
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | graph_placer_test.py | 117 available_devices = [device.name for device in gcluster.ListDevices()]
134 available_devices = [device.name for device in gcluster.ListDevices()]
|
| D | cluster.py | 71 def ListDevices(self): member in Cluster
|
| D | controller.py | 56 self.devices = cluster.ListDevices()
|
| /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/python/client/ |
| D | session_ref.h | 46 Status ListDevices(std::vector<DeviceAttributes>* response) override;
|
| D | session_ref.cc | 246 RUN_WITH_TIMESTAMP(ListDevices, response); in RecordListDevices()
444 Status SessionRef::ListDevices(std::vector<DeviceAttributes>* response) { in ListDevices() function in tensorflow::SessionRef
445 LOG_AND_RUN_OPERATION(ListDevices, response); in ListDevices()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | fused_batchnorm_reserve_space_test.cc | 46 TF_RETURN_IF_ERROR(session->ListDevices(&devices)); in GetTestDevice()
|
| /external/tensorflow/tensorflow/cc/profiler/ |
| D | profiler_test.cc | 128 TF_CHECK_OK(session->ListDevices(&resp)); in TEST_F()
|
| /external/tensorflow/tensorflow/core/public/ |
| D | session.h | 214 virtual Status ListDevices(std::vector<DeviceAttributes>* response) = 0;
|