Home
last modified time | relevance | path

Searched refs:shard_count (Results 1 – 25 of 86) sorted by relevance

1234

/external/tensorflow/tensorflow/python/kernel_tests/linalg/
DBUILD46 shard_count = 5,
84 shard_count = 8,
106 shard_count = 8,
128 shard_count = 5,
150 shard_count = 10,
173 shard_count = 5,
195 shard_count = 5,
217 shard_count = 5,
238 shard_count = 5,
260 shard_count = 5,
[all …]
/external/tensorflow/tensorflow/core/platform/default/
Ddistribute.bzl17 shard_count = 1,
36 shard_count: number of shards to split the tests across.
58 shard_count = shard_count,
73 shard_count = shard_count,
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
DBUILD95 shard_count = 4,
115 shard_count = 4,
139 shard_count = 4,
160 shard_count = 4,
181 shard_count = 4,
200 shard_count = 4,
220 shard_count = 4,
239 shard_count = 8,
265 shard_count = 4,
297 shard_count = 2,
/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/
DBUILD42 shard_count = 10,
58 shard_count = 50,
69 shard_count = 50,
80 shard_count = 50,
/external/tensorflow/tensorflow/python/keras/saving/
DBUILD87 shard_count = 8,
104 shard_count = 4,
118 shard_count = 4,
137 shard_count = 4,
153 shard_count = 4,
171 shard_count = 4,
208 shard_count = 8,
/external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/
DBUILD42 shard_count = 4,
60 shard_count = 7,
82 shard_count = 4,
/external/tensorflow/tensorflow/python/keras/layers/
DBUILD484 shard_count = 3,
501 shard_count = 8,
516 shard_count = 8,
549 shard_count = 4,
570 shard_count = 8,
588 shard_count = 3,
602 shard_count = 3,
673 shard_count = 4,
689 shard_count = 4,
716 shard_count = 4,
[all …]
/external/brotli/java/org/brotli/wrapper/enc/
DBUILD53 shard_count = 15,
69 shard_count = 15,
85 shard_count = 15,
/external/tensorflow/tensorflow/python/keras/tests/
DBUILD29 shard_count = 4,
45 shard_count = 4,
94 shard_count = 4,
118 shard_count = 4,
150 shard_count = 16,
180 shard_count = 16,
202 shard_count = 4,
221 shard_count = 4,
264 shard_count = 8,
349 shard_count = 20,
/external/tensorflow/tensorflow/python/keras/distribute/
DBUILD156 shard_count = 4,
227 shard_count = 10,
355 shard_count = 4,
372 shard_count = 20,
445 shard_count = 19,
467 shard_count = 8,
488 shard_count = 16,
547 shard_count = 31,
567 shard_count = 7,
588 shard_count = 4,
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
DBUILD49 shard_count = 20,
242 shard_count = 5,
584 shard_count = 5,
600 shard_count = 5,
764 shard_count = 16,
857 shard_count = 3,
997 shard_count = 10,
1353 shard_count = 10,
1571 shard_count = 6,
1640 shard_count = 10,
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/random/
DBUILD82 shard_count = 3,
128 shard_count = 10,
147 shard_count = 4,
189 shard_count = 3,
/external/tensorflow/tensorflow/python/keras/premade/
DBUILD41 shard_count = 2,
55 shard_count = 2,
/external/tensorflow/tensorflow/python/keras/engine/
DBUILD276 shard_count = 4,
334 shard_count = 2,
397 shard_count = 20,
455 shard_count = 4,
492 shard_count = 6,
523 shard_count = 30,
592 shard_count = 3,
611 shard_count = 8,
649 shard_count = 3,
667 shard_count = 8,
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/signal/
DBUILD28 shard_count = 16,
43 shard_count = 8,
152 shard_count = 4,
/external/tensorflow/tensorflow/python/keras/applications/
DBUILD60 shard_count = 36,
271 shard_count = 3,
290 shard_count = 8,
341 shard_count = 2,
/external/tensorflow/tensorflow/compiler/tests/
DBUILD205 shard_count = 5,
234 shard_count = 2,
499 shard_count = 10,
522 shard_count = 5,
543 shard_count = 5,
698 shard_count = 6,
774 shard_count = 10,
944 shard_count = 20,
964 shard_count = 20,
1022 shard_count = 5,
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
DBUILD369 shard_count = 50,
390 shard_count = 40,
414 shard_count = 50,
569 shard_count = 30,
628 shard_count = 2,
666 shard_count = 32,
732 shard_count = 25,
799 shard_count = 48,
820 shard_count = 48,
843 shard_count = 48,
[all …]
/external/rust/crates/grpcio-sys/grpc/bazel/
Dgrpc_build_system.bzl178 …ut = None, tags = [], exec_compatible_with = [], exec_properties = {}, shard_count = None, flaky =…
196 "shard_count": shard_count,
228 shard_count = shard_count,
/external/tensorflow/tensorflow/python/keras/
DBUILD455 shard_count = 8,
504 shard_count = 4,
535 shard_count = 4,
558 shard_count = 4,
572 shard_count = 6,
613 shard_count = 8,
631 shard_count = 4,
/external/tensorflow/tensorflow/lite/kernels/internal/
DBUILD909 shard_count = 2,
929 shard_count = 2,
946 shard_count = 2,
962 shard_count = 2,
1003 shard_count = 4,
1020 shard_count = 4,
1033 shard_count = 4,
1055 shard_count = 1,
1070 shard_count = 1,
/external/tensorflow/tensorflow/python/distribute/coordinator/
DBUILD41 shard_count = 50,
72 shard_count = 27,
/external/angle/src/tests/restricted_traces/
Drestricted_trace_gold_tests.py412 def _shard_tests(tests, shard_count, shard_index): argument
413 return [tests[index] for index in range(shard_index, len(tests), shard_count)]
455 args.shard_count = int(env.pop('GTEST_TOTAL_SHARDS'))
488 sharded_tests = _shard_tests(tests['traces'], args.shard_count, args.shard_index)
/external/google-fruit/tests/
Dbuild_defs.bzl18 shard_count = 32,
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc325 int64 shard_count = rhs.sharding().tile_assignment().dim(rhs_dimension); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
332 CeilOfRatio(lhs.base_shape().dimensions(lhs_dimension), shard_count); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
334 CeilOfRatio(rhs.base_shape().dimensions(rhs_dimension), shard_count); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
335 shard_counts[i] = shard_count; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
633 int64 shard_count = lhs.sharding().tile_assignment().dim(lhs_dimension); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local
641 CeilOfRatio(lhs.base_shape().dimensions(lhs_dimension), shard_count); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
643 CeilOfRatio(rhs.base_shape().dimensions(rhs_dimension), shard_count); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
644 shard_counts[i] = shard_count; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()

1234