Home
last modified time | relevance | path

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

12

/external/autotest/client/common_lib/cros/manual/
Dcfm_helper.py111 for pid_vid, device_count in speaker_list.iteritems():
112 if device_count > 0:
113 peripheral_map[pid_vid] = device_count
115 for pid_vid, device_count in camera_list.iteritems():
116 if device_count > 0:
117 peripheral_map[pid_vid] = device_count
119 for pid_vid, device_count in controller_list.iteritems():
120 if device_count > 0:
121 peripheral_map[pid_vid] = device_count
123 for pid_vid, device_count in display_list.iteritems():
[all …]
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_testlib.cc43 auto iter = options.config.device_count().find("CPU"); in MakeTestCluster()
44 if (iter != options.config.device_count().end()) { in MakeTestCluster()
47 iter = options.config.device_count().find("GPU"); in MakeTestCluster()
48 if (iter != options.config.device_count().end()) { in MakeTestCluster()
/external/tensorflow/tensorflow/compiler/xla/service/
Dplatform_util.cc209 int device_count = platform->VisibleDeviceCount(); in GetStreamExecutors() local
210 if (device_count <= 0) { in GetStreamExecutors()
218 device_count = 1; in GetStreamExecutors()
220 std::vector<se::StreamExecutor*> stream_executors(device_count, nullptr); in GetStreamExecutors()
224 tensorflow::Env::Default(), "device_initialization", device_count); in GetStreamExecutors()
225 for (int i = 0; i < device_count; ++i) { in GetStreamExecutors()
Dbackend.h92 int device_count() const { return stream_executors_.size(); } in device_count() function
132 return (device_ordinal >= 0 && device_ordinal < device_count() && in device_ordinal_supported()
Dservice.cc155 if (execute_backend_->device_count() > 0) { in Service()
156 CHECK_GE(execute_backend_->device_count(), options_.number_of_replicas()) in Service()
162 for (int i = 0; i < execute_backend_->device_count(); ++i) { in Service()
706 execute_backend_->device_count()) { in ExecuteParallel()
813 const int64 available_device_count = execute_backend_->device_count(); in GetDeviceHandles()
818 if (available_device_count < arg->device_count() * replica_count) { in GetDeviceHandles()
822 arg->device_count(), available_device_count); in GetDeviceHandles()
825 for (int64 i = 0; i < arg->device_count(); ++i) { in GetDeviceHandles()
828 device_handle.set_device_count(arg->device_count()); in GetDeviceHandles()
1622 device_handle.device_count())); in Replicas()
/external/tensorflow/tensorflow/stream_executor/
Dplatform.cc96 int device_count = VisibleDeviceCount(); in GetPeerAccessMap() local
97 for (int i = 0; i < device_count; ++i) { in GetPeerAccessMap()
98 for (int j = 0; j < device_count; ++j) { in GetPeerAccessMap()
/external/tensorflow/tensorflow/core/common_runtime/
Dthreadpool_device_factory.cc34 auto iter = options.config.device_count().find("CPU"); in CreateDevices()
35 if (iter != options.config.device_count().end()) { in CreateDevices()
Dprocess_function_library_runtime_test.cc36 auto* device_count = options.config.mutable_device_count(); in Init() local
37 device_count->insert({"CPU", 2}); in Init()
/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/core/distributed_runtime/
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/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/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/core/common_runtime/gpu/
Dgpu_device_factory.cc112 auto iter = options.config.device_count().find("CPU"); in CreateDevices()
113 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}))
Dexporter_test.py63 config=config_pb2.ConfigProto(device_count={"CPU": 2})) as sess:
137 config=config_pb2.ConfigProto(device_count={"CPU": 2})) as sess:
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.cc270 int64 device_count) { in GetDeviceHandles() argument
271 if (device_count < 1) { in GetDeviceHandles()
275 request.set_device_count(device_count); in GetDeviceHandles()
Dlocal_client.cc229 int LocalClient::device_count() const { in device_count() function in xla::LocalClient
230 return local_service_->backend().device_count(); in device_count()
Dlocal_client.h166 int device_count() const;
Dclient.h90 StatusOr<std::vector<DeviceHandle>> GetDeviceHandles(int64 device_count);
/external/tensorflow/tensorflow/python/data/kernel_tests/
Diterator_ops_cluster_test.py44 worker_config.device_count["CPU"] = 2
97 worker_config.device_count["CPU"] = 2
/external/tensorflow/tensorflow/python/kernel_tests/
Dsave_restore_ops_test.py32 target="", config=config_pb2.ConfigProto(device_count={"CPU": 2})):
/external/tensorflow/tensorflow/python/client/
Dtimeline_test.py140 config = config_pb2.ConfigProto(device_count={'CPU': 3})
171 config = config_pb2.ConfigProto(device_count={'CPU': 3})
/external/tensorflow/tensorflow/python/profiler/
Dpprof_profiler.py293 device_count = len(self._run_metadata.step_stats.dev_stats)
296 (device_index + 1, device_count, device_stats.device))
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_allocation_test.cc83 for (int d = 0; d < local_client_->device_count(); ++d) { in XLA_TEST_F()
/external/tensorflow/tensorflow/contrib/nccl/kernels/
Dnccl_manager.cc261 int device_count = num_devices; in GetCommunicator() local
266 cudaGetDeviceCount(&device_count); in GetCommunicator()
269 if (num_devices <= device_count) { in GetCommunicator()

12