Home
last modified time | relevance | path

Searched refs:devices_ (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/core/distributed_runtime/
Dremote_device_test.cc40 std::vector<Device*> devices_; member in tensorflow::RemoteDeviceTest
68 devices_ = *found; in SetUp()
72 EXPECT_EQ(devices_.size(), 2); in SetUp()
73 std::sort(devices_.begin(), devices_.end(), [](Device* a, Device* b) { in SetUp()
79 for (auto d : devices_) delete d; in TearDown()
85 EXPECT_EQ(devices_[0]->name(), strings::StrCat(remote_name_, "/cpu:0")); in TEST_F()
86 EXPECT_EQ(devices_[0]->attributes().device_type(), in TEST_F()
88 EXPECT_EQ(devices_[0]->attributes().memory_limit(), 256 << 20); in TEST_F()
89 EXPECT_EQ(devices_[1]->name(), strings::StrCat(remote_name_, "/cpu:1")); in TEST_F()
90 EXPECT_EQ(devices_[1]->attributes().memory_limit(), 256 << 20); in TEST_F()
Dscheduler.h84 const DeviceSet* devices_; variable
109 const DeviceSet* devices_;
Dscheduler.cc192 : devices_(devices), in GreedyScheduler()
196 for (Device* d : devices_->devices()) { in GreedyScheduler()
296 : devices_(devices), cost_model_(cost_model), graph_(g) {} in PriorityScheduler()
303 GreedyScheduler greedysched(devices_, cost_model_, graph_, &slacks); in ComputeSchedule()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_memory_test.cc28 std::unordered_map<string, DeviceProperties> devices_; member in tensorflow::grappler::__anond85ec95a0111::GraphMemoryTest
32 devices_["/CPU:0"].set_type("CPU"); in GraphMemoryTest()
33 devices_["/CPU:0"].set_num_cores(1); in GraphMemoryTest()
34 devices_["/CPU:0"].set_frequency(1); in GraphMemoryTest()
35 devices_["/CPU:0"].set_bandwidth(1); in GraphMemoryTest()
37 devices_["/GPU:0"].set_type("GPU"); in GraphMemoryTest()
38 devices_["/GPU:0"].set_num_cores(1); in GraphMemoryTest()
39 devices_["/GPU:0"].set_frequency(1); in GraphMemoryTest()
40 devices_["/CPU:0"].set_bandwidth(1); in GraphMemoryTest()
41 (*devices_["/GPU:0"].mutable_environment())["architecture"] = "3"; in GraphMemoryTest()
[all …]
Dvirtual_placer.cc33 devices_ = cluster->GetDevices(); in VirtualPlacer()
34 lfqn_map_.reserve(devices_.size()); in VirtualPlacer()
35 for (const auto& kv : devices_) { in VirtualPlacer()
45 if (devices_.empty()) { in VirtualPlacer()
49 DeviceProperties& prop = devices_["UNKNOWN"]; in VirtualPlacer()
51 } else if (devices_.size() == 1) { in VirtualPlacer()
54 default_device_name_ = devices_.begin()->first; in VirtualPlacer()
87 default_device_name_ = devices_.begin()->first; // Any device. in VirtualPlacer()
121 auto it = devices_.find(device); in get_device()
122 DCHECK(it != devices_.end()); in get_device()
Dvirtual_placer.h47 std::unordered_map<string, DeviceProperties> devices_; variable
/external/tensorflow/tensorflow/core/common_runtime/
Ddevice_mgr.cc29 : devices_(std::move(devices)), name_backing_store_(128) { in DeviceMgr()
30 for (auto& d : devices_) { in DeviceMgr()
64 devices->reserve(devices_.size()); in ListDeviceAttributes()
65 for (const auto& dev : devices_) { in ListDeviceAttributes()
71 std::vector<Device*> devices(devices_.size()); in ListDevices()
72 for (size_t i = 0; i < devices_.size(); ++i) { in ListDevices()
73 devices[i] = devices_[i].get(); in ListDevices()
80 for (const auto& dev : devices_) { in DebugString()
88 for (const auto& dev : devices_) { in DeviceMappingString()
114 for (const auto& dev : devices_) { in ClearContainers()
Ddevice_set.cc34 devices_.push_back(device); in AddDevice()
46 for (Device* d : devices_) { in FindMatchingDevices()
77 for (Device* d : devices_) { in PrioritizedDeviceTypeList()
Ddevice_set_test.cc45 devices_.AddDevice(d); in AddDevice()
49 return devices_.PrioritizedDeviceTypeList(); in types()
53 DeviceSet devices_; member in tensorflow::__anonf0ab42df0111::DeviceSetTest
Dplacer.cc85 devices_(devices), in Placer()
100 if (devices_->devices().empty()) { in Run()
113 ColocationGraph colocation_graph(graph_, devices_, default_device_, in Run()
241 devices_->FindDeviceByName(candidate_device_name); in CanAssignToDevice()
Ddevice_set.h51 const std::vector<Device*>& devices() const { return devices_; } in devices()
75 std::vector<Device*> devices_;
/external/webrtc/talk/media/webrtc/
Dfakewebrtcdeviceinfo.h48 devices_.push_back(Device(device_name, device_id)); in AddDevice()
58 return static_cast<int>(devices_.size()); in NumberOfDevices()
110 return (num < devices_.size()) ? &devices_[num] : NULL; in GetDeviceByIndex()
113 for (size_t i = 0; i < devices_.size(); ++i) { in GetDeviceById()
114 if (devices_[i].id == reinterpret_cast<const char*>(device_id)) { in GetDeviceById()
115 return &devices_[i]; in GetDeviceById()
122 std::vector<Device> devices_;
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item.cc45 item.devices_ = devices_; in WithGraph()
133 return devices_; in devices()
148 devices_.insert(DeviceNameUtils::ParsedNameToString(name)); in AddDevice()
171 VLOG(2) << "Inferred device set: [" << absl::StrJoin(devices_, ", ") << "]"; in InferDevicesFromGraph()
179 void GrapplerItem::ClearDevices() { devices_.clear(); } in ClearDevices()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_testlib.h59 const std::vector<DeviceAttributes>& devices() const { return devices_; } in devices()
66 std::vector<DeviceAttributes> devices_; variable
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.cc59 devices_(device_mgr->ListDevices()), in EagerContext()
98 for (auto* device : devices_) { in InitDeviceMapAndAsync()
106 devices_.push_back(device); in InitDeviceMapAndAsync()
112 for (Device* d : devices_) { in InitDeviceMapAndAsync()
284 for (Device* device : devices_) { in StartStep()
438 devices_ = local_unowned_device_manager_->ListDevices(); in StoreCollectiveOpsServer()
481 devices_ = local_unowned_device_manager_->ListDevices(); in InitializeRemote()
Dcontext.h114 std::vector<Device*>* devices() { return &devices_; } in devices()
143 Device* HostCPU() const { return devices_[0]; } in HostCPU()
264 std::vector<Device*> devices_; variable
/external/tensorflow/tensorflow/core/kernels/data/
Dmulti_device_iterator_ops.cc56 devices_(devices), in MultiDeviceIterator()
65 return strings::StrCat("MultiDeviceIterator for ", devices_.size(), in DebugString()
87 devices_.size(), max_buffer_size, incarnation_id_, std::move(iterator), in Init()
356 const std::vector<string> devices_; member in tensorflow::data::__anon2db6daca0111::MultiDeviceIterator
380 OP_REQUIRES_OK(ctx, ctx->GetAttr("devices", &devices_)); in MultiDeviceIteratorHandleOp()
422 context->env(), output_types_, output_shapes_, devices_, in Compute()
440 output_shapes_, devices_, in Compute()
482 std::vector<string> devices_; member in tensorflow::data::__anon2db6daca0111::MultiDeviceIteratorHandleOp
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager_test.cc68 devices_ = new std::vector<std::unique_ptr<BaseGPUDevice>>(GetGPUDevices()); in SetUpTestCase()
69 LOG(INFO) << "Running test with " << devices_->size() << " gpus"; in SetUpTestCase()
72 static int32 NumGPUs() { return static_cast<int32>(devices_->size()); } in NumGPUs()
74 static void TearDownTestCase() { delete devices_; } in TearDownTestCase()
224 return devices_->at(rank % devices_->size()).get(); in GetDevice()
239 static std::vector<std::unique_ptr<BaseGPUDevice>>* devices_; member in tensorflow::NcclManagerTest
245 std::vector<std::unique_ptr<BaseGPUDevice>>* NcclManagerTest<Scalar>::devices_ = member in tensorflow::NcclManagerTest<Scalar>
/external/tensorflow/tensorflow/core/grappler/clusters/
Dcluster.h95 return devices_; in GetDevices()
132 std::unordered_map<string, DeviceProperties> devices_;
Dcluster.cc100 device_names.reserve(devices_.size()); in GetDeviceNames()
101 for (const auto& device : devices_) { in GetDeviceNames()
Dvirtual_cluster.cc37 devices_ = devices; in VirtualCluster()
54 devices_[device->name()] = props; in VirtualCluster()
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node.h213 exec_.mutable_devices()->Reserve(devices_.size()); in ToProto()
214 for (const string& d : devices_) { in ToProto()
233 devices_.clear(); in FromProto()
234 devices_.insert(exec.devices().begin(), exec.devices().end()); in FromProto()
284 std::set<string> devices_; variable
790 devices_.clear(); in SnapshotNodes()
818 devices_.insert(node->canonical_device()); in SnapshotNodes()
853 const std::set<string>& devices() const { return devices_; } in devices()
901 std::set<string> devices_; variable
Dtfprof_timeline.h116 const std::map<string, Device>& devices() const { return devices_; } in devices()
119 std::map<string, Device> devices_;
/external/tensorflow/tensorflow/tools/optimization/
Doptimization_pass_runner.cc126 for (auto& device : devices_) { in RunMain()
163 devices_.push_back(FakeDevice::Make(name, type)); in AddDevice()
Doptimization_pass_runner.h56 std::vector<std::unique_ptr<Device>> devices_; variable

12