Home
last modified time | relevance | path

Searched refs:device_count (Results 1 – 25 of 83) sorted by relevance

1234

/external/autotest/client/common_lib/cros/manual/
Dcfm_helper.py110 for pid_vid, device_count in speaker_list.iteritems():
111 if device_count > 0:
112 peripheral_map[pid_vid] = device_count
114 for pid_vid, device_count in camera_list.iteritems():
115 if device_count > 0:
116 peripheral_map[pid_vid] = device_count
118 for pid_vid, device_count in controller_list.iteritems():
119 if device_count > 0:
120 peripheral_map[pid_vid] = device_count
122 for pid_vid, device_count in display_list.iteritems():
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultiple_devices_on_host_test.cc55 void TestWithDeviceCount(const int device_count) { in TestWithDeviceCount() argument
74 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount()
85 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount()
91 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount()
99 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount()
104 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount()
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dmulti_device_iterator_test.py48 config = config_pb2.ConfigProto(device_count={"CPU": 3})
59 config = config_pb2.ConfigProto(device_count={"CPU": 3})
78 config = config_pb2.ConfigProto(device_count={"CPU": 2})
97 config = config_pb2.ConfigProto(device_count={"CPU": 3})
121 config = config_pb2.ConfigProto(device_count={"CPU": 3})
149 config = config_pb2.ConfigProto(device_count={"CPU": 3})
178 config = config_pb2.ConfigProto(device_count={"CPU": 3})
207 config = config_pb2.ConfigProto(device_count={"CPU": 3})
239 config = config_pb2.ConfigProto(device_count={"CPU": 2, "GPU": 1})
260 config = config_pb2.ConfigProto(device_count={"CPU": 2, "GPU": 1})
[all …]
/external/skia/src/compute/common/cl/
Dfind_cl.c87 cl_uint device_count; in clFindIdsByName() local
94 &device_count); in clFindIdsByName()
96 cl_device_id * const device_ids = ALLOCA_MACRO(sizeof(*device_ids) * device_count); in clFindIdsByName()
100 device_count, in clFindIdsByName()
107 for (cl_uint jj=0; jj<device_count; jj++) in clFindIdsByName()
/external/skqp/src/compute/common/cl/
Dfind_cl.c87 cl_uint device_count; in clFindIdsByName() local
94 &device_count); in clFindIdsByName()
96 cl_device_id * const device_ids = ALLOCA_MACRO(sizeof(*device_ids) * device_count); in clFindIdsByName()
100 device_count, in clFindIdsByName()
107 for (cl_uint jj=0; jj<device_count; jj++) in clFindIdsByName()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_testlib.cc44 auto iter = options.config.device_count().find("CPU"); in MakeTestCluster()
45 if (iter != options.config.device_count().end()) { in MakeTestCluster()
48 iter = options.config.device_count().find("GPU"); in MakeTestCluster()
49 if (iter != options.config.device_count().end()) { in MakeTestCluster()
/external/tensorflow/tensorflow/compiler/xla/service/
Dplatform_util.cc214 int device_count = platform->VisibleDeviceCount(); in GetStreamExecutors() local
215 if (device_count <= 0) { in GetStreamExecutors()
225 device_count = in GetStreamExecutors()
228 std::vector<se::StreamExecutor*> stream_executors(device_count, nullptr); in GetStreamExecutors()
232 tensorflow::Env::Default(), "device_initialization", device_count); in GetStreamExecutors()
233 for (int i = 0; i < device_count; ++i) { in GetStreamExecutors()
Dbackend.h99 int device_count() const { return stream_executors_.size(); } in device_count() function
136 return (device_ordinal >= 0 && device_ordinal < device_count() && in device_ordinal_supported()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed_test.cc206 const int device_count = num_workers * num_devices; in IssueRequests() local
211 cp_.resize(device_count); in IssueRequests()
212 status_.resize(device_count); in IssueRequests()
223 int device_count = num_workers * num_devices; in IssueRequest() local
236 [this, idx, device_count](const Status& s) { in IssueRequest()
241 if (num_done_ == device_count) { in IssueRequest()
249 int device_count = num_workers * num_devices; in ValidateCollectiveParams() local
252 if (num_done_ < device_count) { in ValidateCollectiveParams()
Drpcbench_test.cc58 auto iter = options.config.device_count().find("CPU"); in MakeGRPCCluster()
59 if (iter != options.config.device_count().end()) { in MakeGRPCCluster()
62 iter = options.config.device_count().find("GPU"); in MakeGRPCCluster()
63 if (iter != options.config.device_count().end()) { in MakeGRPCCluster()
/external/tensorflow/tensorflow/python/ops/
Dcollective_ops_test.py35 config=config_pb2.ConfigProto(device_count={'CPU': 2})) as sess:
56 config = config_pb2.ConfigProto(device_count={'CPU': group_size})
96 config=config_pb2.ConfigProto(device_count={'CPU': 2})) as sess:
119 config=config_pb2.ConfigProto(device_count={'CPU': 2})) as sess:
/external/tensorflow/tensorflow/stream_executor/
Dplatform.cc110 int device_count = VisibleDeviceCount(); in GetPeerAccessMap() local
111 for (int i = 0; i < device_count; ++i) { in GetPeerAccessMap()
112 for (int j = 0; j < device_count; ++j) { in GetPeerAccessMap()
/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_device_factory.cc32 auto iter = options.config.device_count().find("SYCL"); in CreateDevices()
33 if (iter != options.config.device_count().end()) { in CreateDevices()
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dcopy_to_device_test.py57 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
82 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
106 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
131 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
156 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
181 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
211 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
244 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
456 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
485 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
Dprefetch_to_device_test.py53 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
103 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
131 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
179 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
/external/tensorflow/tensorflow/tools/test/
Dgpu_info_lib.py139 device_count = ct.c_int()
140 libcudart.cudaGetDeviceCount(ct.byref(device_count))
142 for i in range(device_count.value):
/external/tensorflow/tensorflow/core/common_runtime/
Dthreadpool_device_factory.cc36 auto iter = options.config.device_count().find("CPU"); in CreateDevices()
37 if (iter != options.config.device_count().end()) { in CreateDevices()
Ddevice_resolver_local_test.cc36 auto* device_count = options.config.mutable_device_count(); in DeviceResolverLocalTest() local
37 device_count->insert({"CPU", NUM_DEVS}); in DeviceResolverLocalTest()
Dcollective_executor_mgr_test.cc38 auto* device_count = options.config.mutable_device_count(); in CollectiveExecutorMgrTest() local
40 device_count->insert({"CPU", NUM_DEVS}); in CollectiveExecutorMgrTest()
Dhierarchical_tree_broadcaster.cc130 int device_count = 0; in InitializeCollectiveParams() local
142 perm.push_back(device_count); in InitializeCollectiveParams()
144 col_params->instance.device_names[device_count] == device_name; in InitializeCollectiveParams()
147 device_count += dev_per_task[ti]; in InitializeCollectiveParams()
Ddevice_factory.cc132 auto iter = options.config.device_count().find(type); in NewDevice()
133 if (iter != options.config.device_count().end()) { in NewDevice()
/external/tensorflow/tensorflow/python/debug/cli/
Dprofile_analyzer_cli.py417 device_count = len(self._run_metadata.step_stats.dev_stats)
418 for index in range(device_count):
433 device_stats.device, index, device_count,
477 self, device_name, device_index, device_count, argument
536 device_index + 1, device_count, device_name)
622 device_count = len(self._run_metadata.step_stats.dev_stats)
623 for index in range(device_count):
/external/tensorflow/tensorflow/compiler/jit/
Dcreate_xla_launch_op_test.cc60 auto* device_count = options.config.mutable_device_count(); in Init() local
61 device_count->insert({"CPU", 1}); in Init()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device_factory.cc115 auto iter = options.config.device_count().find("CPU"); in CreateDevices()
116 if (iter != options.config.device_count().end()) { in CreateDevices()
/external/tensorflow/tensorflow/contrib/session_bundle/
Dsession_bundle_test.py94 config=config_pb2.ConfigProto(device_count={"CPU": 2}))
122 config=config_pb2.ConfigProto(device_count={"CPU": 2}))
132 config=config_pb2.ConfigProto(device_count={"CPU": 2}))

1234