/external/tensorflow/tensorflow/core/common_runtime/sycl/ |
sycl_device_factory.cc | 32 auto iter = options.config.device_count().find("SYCL"); 33 if (iter != options.config.device_count().end()) {
|
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
grpc_testlib.cc | 44 auto iter = options.config.device_count().find("CPU"); 45 if (iter != options.config.device_count().end()) { 48 iter = options.config.device_count().find("GPU"); 49 if (iter != options.config.device_count().end()) {
|
/external/skia/src/compute/common/cl/ |
find_cl.c | 87 cl_uint device_count; local 94 &device_count); 96 cl_device_id * const device_ids = ALLOCA_MACRO(sizeof(*device_ids) * device_count); 100 device_count, 107 for (cl_uint jj=0; jj<device_count; jj++)
|
/external/skqp/src/compute/common/cl/ |
find_cl.c | 87 cl_uint device_count; local 94 &device_count); 96 cl_device_id * const device_ids = ALLOCA_MACRO(sizeof(*device_ids) * device_count); 100 device_count, 107 for (cl_uint jj=0; jj<device_count; jj++)
|
/external/autotest/client/common_lib/cros/manual/ |
cfm_helper.py | 110 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/ |
multiple_devices_on_host_test.cc | 55 void TestWithDeviceCount(const int device_count) { 56 // Run `device_count` copies of the XLA program built by BuildComputation. 74 for (int device_ordinal = 0; device_ordinal < device_count; 85 for (int device_ordinal = 0; device_ordinal < device_count; 91 for (int device_ordinal = 0; device_ordinal < device_count; 99 for (int device_ordinal = 0; device_ordinal < device_count; 104 for (int device_ordinal = 0; device_ordinal < device_count;
|
local_client_allocation_test.cc | 83 for (int d = 0; d < local_client_->device_count(); ++d) {
|
/external/tensorflow/tensorflow/core/common_runtime/ |
device_resolver_local_test.cc | 36 auto* device_count = options.config.mutable_device_count(); local 37 device_count->insert({"CPU", NUM_DEVS});
|
threadpool_device_factory.cc | 36 auto iter = options.config.device_count().find("CPU"); 37 if (iter != options.config.device_count().end()) {
|
collective_executor_mgr_test.cc | 38 auto* device_count = options.config.mutable_device_count(); local 40 device_count->insert({"CPU", NUM_DEVS});
|
collective_rma_local_test.cc | 43 auto* device_count = options.config.mutable_device_count(); local 44 device_count->insert({"CPU", NUM_DEVS});
|
device_factory.cc | 132 auto iter = options.config.device_count().find(type); 133 if (iter != options.config.device_count().end()) {
|
/external/tensorflow/tensorflow/stream_executor/ |
platform.cc | 110 int device_count = VisibleDeviceCount(); local 111 for (int i = 0; i < device_count; ++i) { 112 for (int j = 0; j < device_count; ++j) {
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
platform_util.cc | 214 int device_count = platform->VisibleDeviceCount(); local 215 if (device_count <= 0) { 225 device_count = 228 std::vector<se::StreamExecutor*> stream_executors(device_count, nullptr); 232 tensorflow::Env::Default(), "device_initialization", device_count); local 233 for (int i = 0; i < device_count; ++i) {
|
backend.h | 99 int device_count() const { return stream_executors_.size(); } function in class:xla::Backend 136 return (device_ordinal >= 0 && device_ordinal < device_count() &&
|
/external/tensorflow/tensorflow/python/data/kernel_tests/ |
multi_device_iterator_test.py | 48 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/tensorflow/tensorflow/compiler/jit/ |
create_xla_launch_op_test.cc | 60 auto* device_count = options.config.mutable_device_count(); local 61 device_count->insert({"CPU", 1});
|
/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
gpu_device_factory.cc | 115 auto iter = options.config.device_count().find("CPU"); 116 if (iter != options.config.device_count().end()) {
|
/external/tensorflow/tensorflow/core/distributed_runtime/ |
collective_param_resolver_distributed_test.cc | 206 const int device_count = num_workers * num_devices; local 211 cp_.resize(device_count); 212 status_.resize(device_count); 223 int device_count = num_workers * num_devices; local 236 [this, idx, device_count](const Status& s) { 241 if (num_done_ == device_count) { 249 int device_count = num_workers * num_devices; local 252 if (num_done_ < device_count) {
|
rpcbench_test.cc | 58 auto iter = options.config.device_count().find("CPU"); 59 if (iter != options.config.device_count().end()) { 62 iter = options.config.device_count().find("GPU"); 63 if (iter != options.config.device_count().end()) {
|
/external/tensorflow/tensorflow/python/ops/ |
collective_ops_test.py | 35 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/compiler/xla/client/ |
local_client.h | 172 int device_count() const;
|
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
prefetch_to_device_test.py | 53 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})
|
copy_to_device_test.py | 57 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} [all...] |
/external/tensorflow/tensorflow/tools/test/ |
gpu_info_lib.py | 139 device_count = ct.c_int() 140 libcudart.cudaGetDeviceCount(ct.byref(device_count)) 142 for i in range(device_count.value):
|