/external/tensorflow/tensorflow/core/distributed_runtime/ |
D | remote_device_test.cc | 40 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()
|
D | scheduler.h | 84 const DeviceSet* devices_; variable 109 const DeviceSet* devices_;
|
D | scheduler.cc | 192 : 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/ |
D | graph_memory_test.cc | 28 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 …]
|
D | virtual_placer.cc | 33 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()
|
D | virtual_placer.h | 47 std::unordered_map<string, DeviceProperties> devices_; variable
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | device_mgr.cc | 29 : 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()
|
D | device_set.cc | 34 devices_.push_back(device); in AddDevice() 46 for (Device* d : devices_) { in FindMatchingDevices() 77 for (Device* d : devices_) { in PrioritizedDeviceTypeList()
|
D | device_set_test.cc | 45 devices_.AddDevice(d); in AddDevice() 49 return devices_.PrioritizedDeviceTypeList(); in types() 53 DeviceSet devices_; member in tensorflow::__anonf0ab42df0111::DeviceSetTest
|
D | placer.cc | 85 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()
|
D | device_set.h | 51 const std::vector<Device*>& devices() const { return devices_; } in devices() 75 std::vector<Device*> devices_;
|
/external/webrtc/talk/media/webrtc/ |
D | fakewebrtcdeviceinfo.h | 48 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/ |
D | grappler_item.cc | 45 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/ |
D | grpc_testlib.h | 59 const std::vector<DeviceAttributes>& devices() const { return devices_; } in devices() 66 std::vector<DeviceAttributes> devices_; variable
|
/external/tensorflow/tensorflow/core/common_runtime/eager/ |
D | context.cc | 59 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()
|
D | context.h | 114 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/ |
D | multi_device_iterator_ops.cc | 56 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/ |
D | nccl_manager_test.cc | 68 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/ |
D | cluster.h | 95 return devices_; in GetDevices() 132 std::unordered_map<string, DeviceProperties> devices_;
|
D | cluster.cc | 100 device_names.reserve(devices_.size()); in GetDeviceNames() 101 for (const auto& device : devices_) { in GetDeviceNames()
|
D | virtual_cluster.cc | 37 devices_ = devices; in VirtualCluster() 54 devices_[device->name()] = props; in VirtualCluster()
|
/external/tensorflow/tensorflow/core/profiler/internal/ |
D | tfprof_node.h | 213 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
|
D | tfprof_timeline.h | 116 const std::map<string, Device>& devices() const { return devices_; } in devices() 119 std::map<string, Device> devices_;
|
/external/tensorflow/tensorflow/tools/optimization/ |
D | optimization_pass_runner.cc | 126 for (auto& device : devices_) { in RunMain() 163 devices_.push_back(FakeDevice::Make(name, type)); in AddDevice()
|
D | optimization_pass_runner.h | 56 std::vector<std::unique_ptr<Device>> devices_; variable
|