| /tools/tradefederation/core/src/com/android/tradefed/invoker/shard/ |
| IShardHelper.java | 16 package com.android.tradefed.invoker.shard;
26 * Shard and reschedule the shard if possible.
|
| ShardBuildCloner.java | 16 package com.android.tradefed.invoker.shard;
26 * Helper class that handles cloning a build info from the command line. Shard will get the build
|
| TestsPoolPoller.java | 16 package com.android.tradefed.invoker.shard;
|
| ShardHelper.java | 16 package com.android.tradefed.invoker.shard;
53 * List of the list configuration obj that should be clone to each shard in order to avoid state
66 * Attempt to shard the configuration into sub-configurations, to be re-scheduled to run on
69 * <p>A successful shard action renders the current config empty, and invocation should not
91 // shard this invocation!
104 // When shardCount is available only create 1 poller per shard
197 * Attempt to shard given {@link IRemoteTest}.
200 * @param test the {@link IRemoteTest} to shard
201 * @param shardCount attempted number of shard, can be null.
212 // inject device and build since they might be required to shard
[all...] |
| StrictShardHelper.java | 16 package com.android.tradefed.invoker.shard;
54 throw new RuntimeException("shard-count is null while shard-index is " + shardIndex);
84 "%s is not shardable; the whole test will run in shard 0",
102 * @param shardCount the shard count hint to be provided to some tests.
126 // since each shard needs to be independent.
133 // test did not shard so we add it as is.
148 * consistent. It is acceptable to return an empty list if no tests can be run in the shard.
156 * @param shardCount the total number of shard that need to run.
157 * @return a list of list {@link IRemoteTest}s that have been assigned to each shard. The lis
[all...] |
| /external/autotest/cli/ |
| shard.py | 5 The shard module contains the objects and methods used to
9 create: creates shard
10 remove: deletes shard(s)
12 add_boards: add boards to a given shard
13 remove_board: remove board from a given shard
22 class shard(topic_common.atest): class in inherits:topic_common.atest
23 """shard class
24 atest shard [create|delete|list|add_boards|remove_board] <options>"""
26 topic = msg_topic = 'shard'
31 shard actions""
[all...] |
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| tpu.py | 391 def shard(computation, function
406 shard in parallel.
414 ... = shard(computation, ...)
421 Otherwise, each output is taken from an arbitrary shard.
427 shard of the input.
429 list). Each input tensor has a corresponding shard axes, given
433 input_shard_axes: A list of dimensions along which to shard `inputs`, or
434 `None`. `None` means "shard all inputs along dimension 0". If not `None`,
439 from an arbitrary shard. If the argument is a boolean, the argument's
450 `DeviceAssignment` may be omitted if each shard of the computation use
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| population_count_op.cc | 102 auto shard = [input_ptr, output_ptr](int64 start, int64 limit) { local
118 Shard(worker_threads.num_threads, worker_threads.workers, total_shards,
119 shard_cost, shard);
|
| record_yielder.cc | 71 struct RecordYielder::Shard {
72 int index; // Shard index.
73 std::vector<string> filenames; // File names given to this shard.
74 Notification done; // Notified when this shard is done.
75 Status status; // Shard status.
132 // Shards files and use one thread to go through each shard.
134 std::vector<Shard> shards(N);
136 Shard* shard = &shards[i]; local
137 shard->index = i
[all...] |
| candidate_sampler_ops.cc | 173 int64 shard; local
174 OP_REQUIRES_OK(context, context->GetAttr("shard", &shard));
179 num_shards, shard));
182 num_reserved_ids, num_shards, shard));
|
| compare_and_bitpack_op.cc | 148 auto shard = [&, thresh](int64 start, int64 limit) { local
160 Shard(worker_threads.num_threads, worker_threads.workers, total_shards,
161 shard_cost, shard);
|
| linalg_ops_common.cc | 101 auto shard = [this, &inputs, &input_matrix_shapes, &outputs, local
109 Shard(worker_threads.num_threads, worker_threads.workers,
110 batch_shape.num_elements(), GetCostPerUnit(input_matrix_shapes), shard);
|
| avgpooling_op.cc | 290 auto shard = [context, out_backprop_ptr, input_backprop_ptr, variable
342 Shard(worker_threads.num_threads, worker_threads.workers,
343 out_backprop_batch, shard_cost, shard);
|
| depthwise_conv_op.cc | 185 // Computes one shard of depthwise conv2d output.
186 auto shard = [&ctx, &args, &input, &filter_data, &output, data_format]( local
233 // TODO(andydavis): Estimate shard cost (in cycles) based on the number of
234 // flops/loads/stores required to compute one shard.
238 Shard(worker_threads.num_threads, worker_threads.workers, total_shards,
239 shard_cost, shard);
|
| lrn_op.cc | 328 auto shard = [this, activations, in_shaped, grads_shaped, out_shaped, local
370 Shard(worker_threads.num_threads, worker_threads.workers, nodes * batch,
371 depth * depth, shard);
|
| pooling_ops_3d.cc | 576 auto shard = [¶ms, &in_mat, &out_mat, &top_diff_mat, &bottom_diff_mat]( local
650 Shard(worker_threads.num_threads, worker_threads.workers,
651 params.tensor_in_batch, shard_cost, shard);
[all...] |
| pooling_ops_common.h | 197 auto shard = [¶ms, &in_mat, &out_mat](int64 start, int64 limit) { local
250 // TODO(andydavis) Consider a higher resolution shard cost model.
253 Shard(worker_threads.num_threads, worker_threads.workers,
254 params.tensor_in_batch, shard_cost, shard);
444 auto shard = [¶ms, &in_mat, &out_mat](int64 start, int64 limit) { local
497 // TODO(andydavis) Consider a higher resolution shard cost model.
500 Shard(worker_threads.num_threads, worker_threads.workers,
501 params.tensor_in_batch, shard_cost, shard);
523 auto shard = [¶ms, &in_flat, &out_flat](int64 start, int64 limit) { local
602 Shard(worker_threads.num_threads, worker_threads.workers
[all...] |
| /tools/tradefederation/core/src/com/android/tradefed/testtype/ |
| VersionedTfLauncher.java | 94 mCmdArgs.add("--shard-count");
96 mCmdArgs.add("--shard-index");
119 IRemoteTest shard = new VersionedTfLauncher(shardCount, shardIndex); local
121 OptionCopier.copyOptions(this, shard);
126 return shard;
|
| /external/autotest/scheduler/shard/ |
| shard_client.py | 40 Autotest shard client
42 The shard client can be run as standalone service. It periodically polls the
46 A shard is set up (by a human) and pointed to the global AFE (cautotest).
47 On the shard, this script periodically makes so called heartbeat requests to the
50 1. Find the previously created (with atest) record for the shard. Shards are
58 3. Find labels that have been assigned to this shard.
81 2. monitor_db on the shard will pick up these jobs and schedule them on the
87 - the shard_id will be set back to the shard's id, so the record won't be
193 job_models.update(shard=None)
214 def shard(self) member in class:ShardClient
[all...] |
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| dataset_ops.py | 358 def shard(self, num_shards, index): member in class:Dataset
368 d = d.shard(FLAGS.num_workers, FLAGS.worker_index)
376 - Be sure to shard before you use any randomizing operator (such as
378 - Generally it is best if the shard operator is used early in the dataset
379 pipeline. For example, when reading from a set of TFRecord files, shard
386 d = d.shard(FLAGS.num_workers, FLAGS.worker_index)
409 return Dataset(self._dataset.shard(num_shards, index))
|
| /tools/tradefederation/core/tests/src/com/android/tradefed/invoker/shard/ |
| TestsPoolPollerTest.java | 16 package com.android.tradefed.invoker.shard;
|
| ShardHelperTest.java | 16 package com.android.tradefed.invoker.shard;
66 "<configuration description=\"shard config test\">\n"
109 * Tests that when --shard-count is given to local sharding we create shard-count number of
110 * shards and not one shard per IRemoteTest.
116 // shard-count is the number of shards we are requesting
117 setter.setOptionValue("shard-count", "3");
128 // Ensure that we did split 1 tests per shard rescheduled.
141 /** Tests that when --shard-count is not provided we create one shard per IRemoteTest. *
[all...] |
| /external/compiler-rt/lib/sanitizer_common/tests/ |
| sanitizer_thread_registry_test.cc | 163 uptr shard; // started from 1. member in struct:__sanitizer::__anon17203::RunThreadArgs
170 uptr shard = (uptr)arg; local
171 num_joined[shard]++;
174 uptr shard = (uptr)arg; local
175 num_started[shard]++;
178 uptr shard = (uptr)arg; local
179 num_created[shard]++;
190 args->registry->CreateThread(0, false, 0, (void*)args->shard));
192 args->registry->StartThread(tids[i], 0, (void*)args->shard);
196 args->registry->JoinThread(tids[i], (void*)args->shard);
[all...] |
| /prebuilts/go/darwin-x86/test/ |
| run.go | 46 shard = flag.Int("shard", 0, "shard index to run. Only applicable if -shards is non-zero.")
176 return int(h.Sum32()%uint32(*shards)) == *shard
44 shard = flag.Int("shard", 0, "shard index to run. Only applicable if -shards is non-zero.") var
|
| /prebuilts/go/linux-x86/test/ |
| run.go | 46 shard = flag.Int("shard", 0, "shard index to run. Only applicable if -shards is non-zero.")
176 return int(h.Sum32()%uint32(*shards)) == *shard
44 shard = flag.Int("shard", 0, "shard index to run. Only applicable if -shards is non-zero.") var
|
