Home
last modified time | relevance | path

Searched refs:Shard (Results 1 – 25 of 72) sorted by relevance

123

/external/tensorflow/tensorflow/core/util/
Dwork_sharder_test.cc37 Shard(num_workers, threads, total, cost_per_unit, in RunSharding()
64 TEST(Shard, Basic) { in TEST() argument
78 TEST(Shard, OverflowTest) { in TEST() argument
84 Shard(workers, &threads, total_elements, cost_per_unit, in TEST()
98 Shard(arg - 1, &threads, total, 1, work); in BM_Sharding()
Dwork_sharder.h59 void Shard(int max_parallelism, thread::ThreadPool* workers, int64 total,
Dwork_sharder.cc32 void Shard(int max_parallelism, thread::ThreadPool* workers, int64 total, in Shard() function
/external/protobuf/java/core/src/test/java/com/google/protobuf/
DIsValidUtf8TestUtil.java167 static final class Shard { class in IsValidUtf8TestUtil
173 public Shard(long index, long start, long lim, long expected) { in Shard() method in IsValidUtf8TestUtil.Shard
215 static final List<Shard> FOUR_BYTE_SHARDS =
218 private static List<Shard> generateFourByteShards(int numShards, long[] expected) {
220 List<Shard> shards = new ArrayList<Shard>(numShards);
225 shards.add(new Shard(i, increment * i, increment * (i + 1), expected[i]));
DIsValidUtf8Test.java41 import com.google.protobuf.IsValidUtf8TestUtil.Shard;
170 for (Shard shard : IsValidUtf8TestUtil.FOUR_BYTE_SHARDS) { in testShardsHaveExpectedRoundTrippables()
/external/tensorflow/tensorflow/core/kernels/
Drecord_yielder.cc71 struct RecordYielder::Shard { struct in tensorflow::RecordYielder
134 std::vector<Shard> shards(N); in MainLoop()
136 Shard* shard = &shards[i]; in MainLoop()
198 void RecordYielder::ShardLoop(Shard* shard) { in ShardLoop()
Drecord_yielder.h152 struct Shard;
153 void ShardLoop(Shard* shard);
Ddiag_op.cc142 Shard(worker_threads.num_threads, worker_threads.workers, size, 5 * size, in operator ()()
162 Shard(worker_threads.num_threads, worker_threads.workers, size, 5, in operator ()()
Dsdca_ops.cc228 Shard(worker_threads.num_threads, worker_threads.workers, in DoCompute()
285 Shard(worker_threads.num_threads, worker_threads.workers, in Compute()
Dsdca_internal.cc461 Shard(worker_threads.num_threads, worker_threads.workers, num_sparse_features, in CreateSparseFeatureRepresentation()
493 Shard(worker_threads.num_threads, worker_threads.workers, num_dense_features, in CreateDenseFeatureRepresentation()
541 Shard(worker_threads.num_threads, worker_threads.workers, num_examples, in ComputeSquaredNormPerExample()
Dnth_element_op.cc119 Shard(worker_threads.num_threads, worker_threads.workers, num_rows, in operator ()()
Dpopulation_count_op.cc118 Shard(worker_threads.num_threads, worker_threads.workers, total_shards, in operator ()()
Dpooling_ops_common.h258 Shard(worker_threads.num_threads, worker_threads.workers, in SpatialMaxPool()
511 Shard(worker_threads.num_threads, worker_threads.workers,
613 Shard(worker_threads.num_threads, worker_threads.workers,
Droll_op.cc171 Shard(worker_threads->num_threads, worker_threads->workers, num_elements, in DoRoll()
301 Shard(worker_threads->num_threads, worker_threads->workers, total_work, in DoRollWithMemcpy()
Dlu_op.cc123 Shard(worker_threads.num_threads, worker_threads.workers, in Compute()
/external/autotest/server/cros/tradefed/tradefed_utils_unittest_data/
DCtsDeqpTestCases-trimmed-inaccurate.txt22 Shard 0 used: [chromeos6-row4-rack16-host9:22]
23 Shard 1 used: [chromeos6-row4-rack16-host17:22]
/external/autotest/frontend/afe/
Drpc_interface_unittest.py231 models.Shard.DoesNotExist, models.Shard.objects.get, pk=shard1.id)
422 shard1 = models.Shard.objects.create(hostname='shard1')
443 shard1 = models.Shard.objects.create(hostname='shard1')
555 shard = models.Shard.objects.create(hostname='test_shard')
566 shard1 = models.Shard.smart_get(shard.id)
643 shard = models.Shard.objects.create(hostname=shard_hostname)
999 shard = models.Shard.objects.create(hostname='shard')
1139 shard = models.Shard.objects.create(hostname=shard_hostname)
1201 shard1 = models.Shard.objects.create(hostname='shard1')
1205 shard2 = models.Shard.objects.create(hostname='shard2')
[all …]
Drpc_interface.py2102 shards = models.Shard.query_objects(filter_data)
2140 shard = models.Shard.objects.get(labels=label)
2143 except models.Shard.DoesNotExist:
2165 shard = models.Shard.add_object(hostname=hostname)
2184 shard = models.Shard.objects.get(hostname=hostname)
2203 shard = models.Shard.objects.get(hostname=hostname)
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dquantile_ops.cc206 Shard(worker_threads.num_threads, worker_threads.workers, num_features_, in Compute()
265 Shard(worker_threads.num_threads, worker_threads.workers, num_features_, in Compute()
332 Shard(worker_threads.num_threads, worker_threads.workers, num_streams, in Compute()
379 Shard(worker_threads.num_threads, worker_threads.workers, num_features_, in Compute()
431 Shard(worker_threads.num_threads, worker_threads.workers, num_streams, in Compute()
490 Shard(worker_threads.num_threads, worker_threads.workers, num_streams, in Compute()
567 Shard(worker_threads.num_threads, worker_threads.workers, num_features_, in Compute()
Dprediction_ops.cc186 Shard(worker_threads->NumThreads(), worker_threads, batch_size, in Compute()
276 Shard(worker_threads->NumThreads(), worker_threads, batch_size, in Compute()
392 Shard(worker_threads->NumThreads(), worker_threads, batch_size, in Compute()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_cholesky_op.cc110 Shard(worker_threads.num_threads, worker_threads.workers, batch_size, in Compute()
189 Shard(worker_threads.num_threads, worker_threads.workers, batch_size, in Compute()
Dsparse_ordering_amd_op.cc97 Shard( in Compute()
/external/tensorflow/tensorflow/core/kernels/tensor_forest/
Dprediction_ops.cc72 Shard(num_threads, worker_threads->workers, batch_size, cost_per_traverse, in Compute()
/external/tensorflow/tensorflow/core/protobuf/
Dsaver.proto25 // Shard the save files, one per device that has Variable nodes.
/external/tensorflow/tensorflow/python/keras/distribute/
DBUILD179 # Shard count is set to an odd number to distribute tasks across
235 # Shard count is set to an odd number to distribute tasks across

123