Home
last modified time | relevance | path

Searched refs:device_names (Results 1 – 25 of 68) sorted by relevance

123

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dimplementation_selector_test.cc67 AttrValue device_names; in TEST_F() local
68 device_names.mutable_list()->add_s("CPU"); in TEST_F()
69 device_names.mutable_list()->add_s("GPU"); in TEST_F()
71 {NDef("x", "DeviceIndex", {}, {{"device_names", device_names}}, in TEST_F()
91 AttrValue device_names; in TEST_F() local
92 device_names.mutable_list()->add_s("CPU"); in TEST_F()
93 device_names.mutable_list()->add_s("GPU"); in TEST_F()
95 {NDef("x", "DeviceIndex", {}, {{"device_names", device_names}}, in TEST_F()
115 AttrValue device_names; in TEST_F() local
116 device_names.mutable_list()->add_s("CPU"); in TEST_F()
[all …]
/external/webrtc/modules/desktop_capture/win/
Dscreen_capturer_win_directx.cc50 const std::vector<std::string>& device_names, in GetScreenListFromDeviceNames() argument
67 for (const auto& device_name : device_names) { in GetScreenListFromDeviceNames()
84 const std::vector<std::string>& device_names) { in GetIndexFromScreenId() argument
86 if (!GetScreenListFromDeviceNames(device_names, &screens)) { in GetIndexFromScreenId()
90 RTC_DCHECK_EQ(device_names.size(), screens.size()); in GetIndexFromScreenId()
186 std::vector<std::string> device_names; in GetSourceList() local
187 if (!controller_->GetDeviceNames(&device_names)) { in GetSourceList()
191 return GetScreenListFromDeviceNames(device_names, sources); in GetSourceList()
200 std::vector<std::string> device_names; in SelectSource() local
201 if (!controller_->GetDeviceNames(&device_names)) { in SelectSource()
[all …]
Dscreen_capturer_win_directx_unittest.cc24 const std::vector<std::string> device_names = { in TEST() local
31 device_names, &screens)); in TEST()
32 ASSERT_EQ(device_names.size(), screens.size()); in TEST()
36 device_names), in TEST()
Dscreen_capture_utils.cc26 std::vector<std::string>* device_names /* = nullptr */) { in GetScreenList() argument
28 if (device_names) { in GetScreenList()
29 RTC_DCHECK_EQ(device_names->size(), 0U); in GetScreenList()
47 if (device_names) { in GetScreenList()
48 device_names->push_back(rtc::ToUtf8(device.DeviceName)); in GetScreenList()
Dscreen_capture_utils_unittest.cc23 std::vector<std::string> device_names; in TEST() local
27 ASSERT_TRUE(GetScreenList(&screens, &device_names)); in TEST()
29 ASSERT_EQ(screens.size(), device_names.size()); in TEST()
Dscreen_capturer_win_directx.h62 const std::vector<std::string>& device_names,
70 const std::vector<std::string>& device_names);
/external/tensorflow/tensorflow/lite/nnapi/
Dnnapi_util.cc46 std::vector<const char*> device_names; in GetDeviceNamesList() local
60 device_names.push_back(buffer); in GetDeviceNamesList()
64 return device_names; in GetDeviceNamesList()
68 std::vector<const char*> device_names = GetDeviceNamesList(); in GetStringDeviceNamesList() local
69 return SimpleJoin(device_names, ","); in GetStringDeviceNamesList()
/external/tensorflow/tensorflow/core/grappler/inputs/
Dtrivial_test_graph_input_yielder.cc33 const std::vector<string>& device_names) { in CreateGraphDef() argument
52 device_names[use_multiple_devices ? j % device_names.size() in CreateGraphDef()
59 device_names[use_multiple_devices ? j % device_names.size() in CreateGraphDef()
91 const std::vector<string>& device_names) in TrivialTestGraphInputYielder() argument
96 device_names_(device_names) {} in TrivialTestGraphInputYielder()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dcluster.cc99 std::vector<string> device_names; in GetDeviceNames() local
100 device_names.reserve(devices_.size()); in GetDeviceNames()
102 device_names.push_back(device.first); in GetDeviceNames()
104 std::sort(device_names.begin(), device_names.end()); in GetDeviceNames()
105 return device_names; in GetDeviceNames()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_supported_ops.cc70 std::vector<string> device_names = XlaOpRegistry::BackendNames(); in SupportedOpsMain() local
71 std::sort(device_names.begin(), device_names.end()); in SupportedOpsMain()
79 absl::StrJoin(device_names, ",")}, in SupportedOpsMain()
/external/autotest/client/common_lib/cros/cfm/
Dcras_node_collector_unittest.py51 device_names = set([node.device_name for node in nodes])
92 device_names = set([node.device_name for node in nodes])
97 self.assertEquals(device_names, set(
/external/tensorflow/tensorflow/python/distribute/v1/
Dall_reduce_test.py94 device_names = []
98 device_names.append(dn)
101 return input_tensors, device_names
106 input_tensors, device_names = self._buildInput(1, 1)
108 output_tensors = ar._build_ring_gather(input_tensors, device_names, 1,
113 input_tensors, device_names = self._buildInput(1, 4)
116 input_tensors, device_names, 2, pred_by_c_d, rank_by_c_d, math_ops.add)
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.cc292 CHECK_EQ(gp.device_names.size(), gp.task_names.size()); in BuildDevRecs()
293 CHECK_EQ(gp.device_names.size(), attributes.size()); in BuildDevRecs()
294 for (int i = 0; i < gp.device_names.size(); ++i) { in BuildDevRecs()
296 DevRec* dr = &tdm[gp.device_names[i]]; in BuildDevRecs()
298 dr->device = gp.device_names[i]; in BuildDevRecs()
486 CHECK_EQ(gp->group_size, gp->device_names.size()); in SortDevicesAndTasks()
495 return gp->device_names[a] < gp->device_names[b]; in SortDevicesAndTasks()
502 new_devs.push_back(gp->device_names[pi]); in SortDevicesAndTasks()
505 gp->device_names = std::move(new_devs); in SortDevicesAndTasks()
513 gr->group.device_names.reserve(gr->devices.size()); in FinishGroup()
[all …]
Ddynamic_device_mgr.cc83 std::vector<StringPiece> device_names; in LookupDevice() local
85 device_names.push_back(itr.first); in LookupDevice()
88 << " all devices: " << absl::StrJoin(device_names, ", "); in LookupDevice()
188 const std::vector<string>& device_names) { in RemoveDevicesByName() argument
190 for (const string& name : device_names) { in RemoveDevicesByName()
Dhierarchical_tree_broadcaster.cc83 col_params->group.device_names[col_params->default_rank]; in InitializeCollectiveParams()
138 participate = col_params->group.device_names[col_params->source_rank] == in InitializeCollectiveParams()
144 col_params->group.device_names[device_count] == device_name; in InitializeCollectiveParams()
167 if (col_params->group.device_names[abs_di] == device_name) { in InitializeCollectiveParams()
419 << col_params_->group.device_names[dst_idx] << " subdiv=" << subdiv in DispatchSend()
422 col_params_->group.device_names[dst_idx], in DispatchSend()
438 << col_params_->group.device_names[src_idx] << " to_device " in DispatchRecv()
442 col_params_->group.device_names[src_idx], in DispatchRecv()
Ddevice_mgr.cc115 std::vector<StringPiece> device_names; in LookupDevice() local
117 device_names.push_back(itr.first); in LookupDevice()
120 << " all devices: " << absl::StrJoin(device_names, ", "); in LookupDevice()
Dcollective_param_resolver_local_test.cc76 EXPECT_EQ(group.device_names, expected_device_order); in RunCompleteDefaultRanking()
101 group.device_names.push_back(strings::StrCat( in TEST_F()
196 ASSERT_EQ(cps[i]->group.device_names.size(), 3); in TEST_F()
200 cps[i]->group.device_names[j]); in TEST_F()
251 ASSERT_EQ(cps[i]->group.device_names.size(), 3); in TEST_F()
255 cps[i]->group.device_names[j]); in TEST_F()
Dcollective_util.cc63 CHECK_GT(col_params.group.device_names.size(), idx); in SubdivPermDebugString()
64 strings::StrAppend(&buf, col_params.group.device_names[idx], "\n"); in SubdivPermDebugString()
/external/webrtc/modules/audio_device/win/
Dcore_audio_base_win.cc282 AudioDeviceNames device_names; in GetDeviceID() local
284 ? core_audio_utility::GetInputDeviceNames(&device_names) in GetDeviceID()
285 : core_audio_utility::GetOutputDeviceNames(&device_names); in GetDeviceID()
287 device_id = device_names[index].unique_id; in GetDeviceID()
319 AudioDeviceNames device_names; in DeviceName() local
320 bool ok = IsInput() ? core_audio_utility::GetInputDeviceNames(&device_names) in DeviceName()
321 : core_audio_utility::GetOutputDeviceNames(&device_names); in DeviceName()
324 if (!ok || static_cast<int>(device_names.size()) <= index) { in DeviceName()
329 *name = device_names[index].device_name; in DeviceName()
332 *guid = device_names[index].unique_id; in DeviceName()
Dcore_audio_utility_win.cc475 webrtc::AudioDeviceNames* device_names) { in GetDeviceNamesInternal() argument
527 device_names->push_back(device_name); in GetDeviceNamesInternal()
532 if (device_names->size() != number_of_active_devices) { in GetDeviceNamesInternal()
540 if (device_names->empty()) { in GetDeviceNamesInternal()
576 device_names->push_front(default_device_name); in GetDeviceNamesInternal()
589 for (size_t i = 0; i < device_names->size(); ++i) { in GetDeviceNamesInternal()
591 << "] friendly name: " << (*device_names)[i].device_name; in GetDeviceNamesInternal()
593 << "] unique id : " << (*device_names)[i].unique_id; in GetDeviceNamesInternal()
798 bool GetInputDeviceNames(webrtc::AudioDeviceNames* device_names) { in GetInputDeviceNames() argument
800 RTC_DCHECK(device_names); in GetInputDeviceNames()
[all …]
/external/tensorflow/tensorflow/python/client/
Dsession_list_devices_test.py75 device_names = set(d.name for d in devices)
77 '/job:worker/replica:0/task:0/device:CPU:0' in device_names)
79 '/job:worker/replica:0/task:1/device:CPU:0' in device_names)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util_test.cc36 bool DeviceNamesToParsedNames(llvm::ArrayRef<std::string> device_names, in DeviceNamesToParsedNames() argument
38 parsed_devices->reserve(device_names.size()); in DeviceNamesToParsedNames()
39 for (const auto& device_name : device_names) { in DeviceNamesToParsedNames()
302 std::vector<std::string> device_names = in TEST() local
304 ASSERT_TRUE(DeviceNamesToParsedNames(device_names, &devices)); in TEST()
318 std::vector<std::string> device_names = in TEST() local
320 ASSERT_TRUE(DeviceNamesToParsedNames(device_names, &devices)); in TEST()
423 std::vector<std::string> device_names = in TEST() local
425 ASSERT_TRUE(DeviceNamesToParsedNames(device_names, &devices)); in TEST()
536 std::vector<std::string> device_names = in TEST() local
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtf_device_index_selector.mlir10 %0 = "tf.DeviceIndex"() {device = "", device_names = ["CPU", "GPU"]} : () -> tensor<i32>
11 %1 = "tf.DeviceIndex"() {device = "", device_names = ["CPU", "GPU"]} : () -> tensor<i32>
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed_test.cc263 EXPECT_EQ(cp_[device_name]->group.device_names.size(), dev_count); in ValidateCollectiveParams()
264 EXPECT_EQ(cp_[device_name]->group.device_names[idx], device_name); in ValidateCollectiveParams()
271 EXPECT_EQ(cp_[dev0]->group.device_names[i], in ValidateCollectiveParams()
272 cp_[device_name]->group.device_names[i]); in ValidateCollectiveParams()
282 for (const string& device_name : cp.group.device_names) { in ValidateDeviceResolver()
/external/tensorflow/tensorflow/compiler/jit/
Ddevice_util_test.cc25 absl::Span<const absl::string_view> device_names, in PickDeviceHelper() argument
29 for (absl::string_view name : device_names) { in PickDeviceHelper()

123