| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| collective_param_resolver_distributed_test.cc | 145 void DefineWorkers(int num_workers, int num_devices,
153 DefineWorker(config, name, device_type, num_devices);
158 const string& device_type, int num_devices) {
160 for (int i = 0; i < num_devices; ++i) {
183 void DefineCollectiveParams(int num_workers, int num_devices) {
188 for (int di = 0; di < num_devices; ++di) {
193 cp.group.group_size = num_workers * num_devices;
205 void IssueRequests(int num_workers, int num_devices) {
206 const int device_count = num_workers * num_devices;
215 for (int di = 0; di < num_devices; ++di)
299 const int num_devices = 1; local
308 const int num_devices = 2; local
317 const int num_devices = 3; local
[all...] |
| /external/blktrace/iowatcher/ |
| tracers.h | 23 int start_blktrace(char **devices, int num_devices, char *trace_name, char *dest);
|
| tracers.c | 52 int start_blktrace(char **devices, int num_devices, char *trace_name, char *dest)
76 if (num_devices == 1) {
86 for (i = 0; i < num_devices; i++) {
|
| /external/mesa3d/src/gallium/state_trackers/clover/api/ |
| context.cpp | 71 cl_uint num_devices; local
77 ret = clGetDeviceIDs(d_platform, type, 0, NULL, &num_devices);
80 devs.resize(num_devices);
81 ret = clGetDeviceIDs(d_platform, type, num_devices, devs.data(), 0);
85 return clCreateContext(d_props, num_devices, devs.data(), pfn_notify,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| nccl_ops.cc | 42 OP_REQUIRES_OK(c, c->GetAttr("num_devices", &num_devices_));
52 int num_devices() const { return num_devices_; } function in class:tensorflow::__anon45434::NcclAsyncOpBase
113 /*num_local_devices=*/num_devices(),
114 /*num_global_devices=*/num_devices(),
145 /*num_local_devices=*/num_devices(),
146 /*num_global_devices=*/num_devices(),
182 /*num_local_devices=*/num_devices(),
183 /*num_global_devices=*/num_devices(),
216 /*num_local_devices=*/num_devices(),
217 /*num_global_devices=*/num_devices(),
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/ |
| xla_sharding.py | 92 def split(cls, tensor, split_dimension, num_devices):
101 num_devices: The number of cores to split `tensor` over.
109 shape[split_dimension] < num_devices):
111 'of splits: shape=%r, dimension=%r, num_devices=%r' %
112 (shape, split_dimension, num_devices))
115 tile_assignment_dims[split_dimension] = num_devices
121 tile_assignment_devices=range(num_devices)))
184 def split(tensor, split_dimension, num_devices):
185 Sharding.split(tensor, split_dimension, num_devices).apply_to_tensor(tensor)
|
| /bionic/libc/kernel/uapi/linux/ |
| pktcdvd.h | 57 __u32 num_devices; member in struct:pkt_ctrl_command
|
| /device/google/bonito/sdm710/kernel-headers/media/ |
| cam_lrme.h | 48 uint32_t num_devices; member in struct:cam_lrme_query_cap_cmd
|
| /device/google/bonito/sdm710/original-kernel-headers/media/ |
| cam_lrme.h | 44 * @num_devices: number of hardware devices
50 uint32_t num_devices; member in struct:cam_lrme_query_cap_cmd
|
| /device/google/crosshatch/sdm845/kernel-headers/media/ |
| cam_lrme.h | 48 uint32_t num_devices; member in struct:cam_lrme_query_cap_cmd
|
| /device/google/crosshatch/sdm845/original-kernel-headers/media/ |
| cam_lrme.h | 44 * @num_devices: number of hardware devices
50 uint32_t num_devices; member in struct:cam_lrme_query_cap_cmd
|
| /external/tensorflow/tensorflow/python/distribute/ |
| all_reduce.py | 196 In the basic ring reduction algorithm there are size(T)/num_devices
317 num_devices = len(input_tensors)
318 if num_devices == 0:
320 if num_devices == 1:
325 num_chunks = num_devices * num_subchunks
326 num_ticks = num_devices - 1
330 for d in range(0, num_devices):
339 for d in range(0, num_devices):
343 seg_index = (rank + num_devices - (2 + tick)) % num_devices
[all...] |
| shared_variable_creator_test.py | 53 num_devices = 3
55 for i in range(num_devices):
|
| cross_device_utils.py | 77 num_devices = len(avail_devices)
80 group_size = num_devices // 2
82 group_0_main_device = i % num_devices
83 group_1_main_device = (group_0_main_device + group_size) % num_devices
184 num_devices = len(devices)
185 if group_size > num_devices:
187 'only %d devices, but group_size=%d' % (num_devices, group_size))
189 num_devices // group_size + (1 if (num_devices % group_size != 0) else 0))
192 groups[i % num_groups].append(devices[i % num_devices])
[all...] |
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorDeviceCuda.h | 60 int num_devices; local
61 cudaError_t status = cudaGetDeviceCount(&num_devices);
68 m_deviceProperties = new cudaDeviceProp[num_devices];
69 for (int i = 0; i < num_devices; ++i) {
119 int num_devices; local
120 cudaError_t err = cudaGetDeviceCount(&num_devices);
123 assert(device < num_devices);
|
| /external/mesa3d/src/loader/ |
| loader.c | 163 int i, num_devices, fd; local
182 num_devices = drmGetDevices2(0, devices, MAX_DRM_DEVICES);
183 if (num_devices < 0)
195 for (i = 0; i < num_devices; i++) {
204 for (i = 0; i < num_devices; i++) {
215 drmFreeDevices(devices, num_devices);
220 drmFreeDevices(devices, num_devices);
|
| /external/webrtc/webrtc/tools/e2e_quality/audio/ |
| audio_e2e_harness.cc | 70 int num_devices = 0; local
71 ASSERT_EQ(0, hardware->GetNumOfPlayoutDevices(num_devices));
76 for (device_index = 0; device_index < num_devices; device_index++) {
|
| /external/kernel-headers/original/uapi/linux/ |
| pktcdvd.h | 101 __u32 num_devices; /* out: Largest device index + 1 */ member in struct:pkt_ctrl_command
|
| /external/tensorflow/tensorflow/python/ops/ |
| nccl_ops.py | 68 num_devices = op.get_attr('num_devices')
75 num_devices=num_devices,
227 num_devices=len(tensors),
|
| /external/tensorflow/tensorflow/core/nccl/ |
| nccl_rewrite.cc | 36 int num_devices = node->num_inputs(); local
41 .Attr("num_devices", num_devices)
121 int num_devices = 0; // Number of distinct devices, incremented below. local
136 recv_index_map[dst_dev] = num_devices;
137 ++num_devices;
142 if (num_devices <= 1) {
164 .Attr("num_devices", num_devices)
|
| /external/webrtc/webrtc/tools/agc/ |
| agc_harness.cc | 158 int num_devices = 0; local
161 RTC_CHECK_EQ(0, hardware_->GetNumOfRecordingDevices(num_devices));
163 for (int i = 0; i < num_devices; i++) {
167 RTC_CHECK_EQ(0, hardware_->GetNumOfPlayoutDevices(num_devices));
169 for (int i = 0; i < num_devices; i++) {
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_sharding.h | 94 Status Validate(const Shape& shape, int64 num_devices) const;
252 Status ValidateTuple(const Shape& shape, int64 num_devices) const;
255 Status ValidateNonTuple(const Shape& shape, int64 num_devices) const;
|
| /external/webrtc/talk/media/devices/ |
| macdevicemanager.cc | 121 size_t num_devices = propsize / sizeof(AudioDeviceID); local
123 new AudioDeviceID[num_devices]);
133 for (size_t i = 0; i < num_devices; ++i) {
|
| /external/mesa3d/include/CL/ |
| cl_dx9_media_sharing.h | 96 cl_uint * num_devices) CL_API_SUFFIX__VERSION_1_2;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| ring_gatherer_test.cc | 133 void Init(int num_workers, int num_devices, DataType dtype,
145 for (int di = 0; di < num_devices; ++di) {
152 int dev_idx = (wi * num_devices) + di;
179 col_params_.group.group_size = num_workers * num_devices;
188 int subdiv_stride = num_devices / num_subdivs;
197 for (int di = 0; di < num_devices; ++di) {
200 for (int di = 0; di < num_devices; ++di) {
202 int other = (di + (is_odd ? 7 : 3)) % num_devices;
213 for (int di = 0; di < num_devices; ++di) {
228 num_devices;
[all...] |
