| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | tpu_cross_replica_ops.cc | 45 int split_count; in __anon173a7a2a0102() local
46 TF_RETURN_IF_ERROR(c->GetAttr("split_count", &split_count)); in __anon173a7a2a0102()
47 if (split_count < 1) { in __anon173a7a2a0102()
48 return errors::InvalidArgument("split_count ", split_count, in __anon173a7a2a0102()
56 (c->Value(num_replicas_per_group) != split_count)) { in __anon173a7a2a0102()
58 "split_count ", split_count, in __anon173a7a2a0102()
88 dims[i] = c->MakeDim(c->Value(dims[i]) * split_count); in __anon173a7a2a0102()
92 (c->Value(dims[i]) % split_count != 0)) { in __anon173a7a2a0102()
95 " not divisible by split_count ", split_count); in __anon173a7a2a0102()
97 dims[i] = c->MakeDim(c->Value(dims[i]) / split_count); in __anon173a7a2a0102()
|
| /aosp_15_r20/external/sdv/vsomeip/third_party/boost/intrusive/test/ |
| D | unordered_test.hpp | 455 BOOST_TEST(testset1.split_count() == BucketSize/2); in test_rehash()
460 BOOST_TEST(testset1.split_count() == (BucketSize/2+1)); in test_rehash()
464 …for(std::size_t split_bucket = testset1.split_count(); split_bucket != BucketSize; ++split_bucket){ in test_rehash()
466 BOOST_TEST(testset1.split_count() == (split_bucket+1)); in test_rehash()
472 BOOST_TEST(testset1.split_count() == BucketSize); in test_rehash()
483 BOOST_TEST(testset1.split_count() == BucketSize); in test_rehash()
494 BOOST_TEST(testset1.split_count() == BucketSize); in test_rehash()
503 BOOST_TEST(testset1.split_count() == BucketSize); in test_rehash()
512 BOOST_TEST(testset1.split_count() == BucketSize); in test_rehash()
544 for(std::size_t split_bucket = testset1.split_count(); split_bucket > 6; --split_bucket){ in test_rehash()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/security/advisory/ |
| H A D | tfsa-2021-176.md | 19 split_count=0)
24 This occurs whenever the `split_count` argument is 0:
27 TF_RETURN_IF_ERROR(c->GetAttr("split_count", &split_count));
31 dims[i] = c->MakeDim(c->Value(dims[i]) / split_count);
|
| /aosp_15_r20/external/rust/android-crates-io/crates/rayon/src/iter/ |
| D | par_bridge.rs | 93 split_count: AtomicUsize::new(num_threads), in drive_unindexed()
103 split_count: AtomicUsize, field
112 let mut count = self.split_count.load(Ordering::SeqCst); in split()
117 match self.split_count.compare_exchange_weak( in split()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/ |
| H A D | api_def_AllToAll.pbtxt | 43 name: "split_count"
51 On each replica, the input is split into `split_count` blocks along
53 receiving `split_count` - 1 blocks from other replicas, we concatenate the
63 split_count=2
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ops/ |
| H A D | tpu_ops.py | 43 split_count, argument
69 split_count=split_count,
84 split_count=op.get_attr("split_count")), None
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | tpu_test.py | 180 split_count=0)
190 split_count=2)
201 split_count=1)
211 split_count=2)
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| H A D | AllToAll.pbtxt | 34 name: "split_count"
87 name: "split_count"
140 name: "split_count"
|
| /aosp_15_r20/external/mesa3d/src/amd/vpelib/src/core/ |
| H A D | resource.c | 326 int32_t split_count, split_idx; in calculate_recout() local
330 split_count = stream_ctx->num_segments - 1; in calculate_recout()
339 if (split_count) { in calculate_recout()
343 int32_t epimo = split_count - (int32_t)data->recout.width % (split_count + 1); in calculate_recout()
345 data->recout.x += ((int32_t)data->recout.width / (split_count + 1)) * split_idx; in calculate_recout()
350 data->recout.width / (uint32_t)(split_count + 1) + (split_idx > epimo ? 1 : 0); in calculate_recout()
|
| /aosp_15_r20/external/sdv/vsomeip/third_party/boost/move/test/ |
| D | bench_merge.cpp | 61 std::size_t split_count = L / 2; in generate_elements() local
62 std::stable_sort(elements.data(), elements.data() + split_count, comp); in generate_elements()
63 std::stable_sort(elements.data() + split_count, elements.data() + L, comp); in generate_elements()
64 return split_count; in generate_elements()
|
| /aosp_15_r20/frameworks/base/tools/aapt2/split/ |
| H A D | TableSplitter.cpp | 181 const size_t split_count = split_constraints_.size(); in SplitTable() local
184 for (size_t idx = 0; idx < split_count; idx++) { in SplitTable()
227 for (size_t idx = 0; idx < split_count; idx++) { in SplitTable()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| H A D | AllToAll.pbtxt | 34 name: "split_count"
87 name: "split_count"
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder.cc | 3078 int64_t concat_dimension, int64_t split_count, in AllToAll() argument
3085 split_count, replica_groups, layout); in AllToAll()
3087 return AllToAllArray(operand, split_dimension, concat_dimension, split_count, in AllToAll()
3092 int64_t concat_dimension, int64_t split_count, in AllToAllArray() argument
3099 concat_dimension, split_count)); in AllToAllArray()
3104 for (int64_t i = 0; i < split_count; ++i) { in AllToAllArray()
3125 sizes.push_back(split_count); in AllToAllArray()
3126 sizes.push_back(operand_shape->dimensions(i) / split_count); in AllToAllArray()
3184 int64_t concat_dimension, int64_t split_count, in AllToAllTuple() argument
3199 concat_dimension, split_count) in AllToAllTuple()
[all …]
|
| H A D | xla_builder.h | 770 int64_t concat_dimension, int64_t split_count,
779 int64_t concat_dimension, int64_t split_count,
1382 int64_t concat_dimension, int64_t split_count,
1389 int64_t concat_dimension, int64_t split_count,
1527 int64_t concat_dimension, int64_t split_count,
2369 int64_t split_count,
2378 int64_t concat_dimension, int64_t split_count,
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | loopnest.cpp | 1573 ExprPtr split_count = alloc<Div>(size, factor_expr); in splitWithTail() local
1590 alloc<Add>(i_tail, alloc<Mul>(split_count, factor_expr)); in splitWithTail()
1608 f->set_start(immLike(split_count, 0)); in splitWithTail()
1609 f->set_stop(split_count); in splitWithTail()
1641 ExprPtr split_count = alloc<Div>( in splitWithMask() local
1674 f->set_start(immLike(split_count, 0)); in splitWithMask()
1675 f->set_stop(split_count); in splitWithMask()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | tpu_strategy.py | 1453 split_count=self._num_replicas_in_sync)
1795 split_count=self.num_replicas_in_sync)
1808 split_count=self.num_replicas_in_sync)
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | shape_inference.cc | 2139 int64_t split_count) { in InferAllToAllShape() argument
2140 TF_RET_CHECK(split_count > 0); in InferAllToAllShape()
2151 if (shape.dimensions(split_dimension) % split_count != 0) { in InferAllToAllShape()
2155 shape.dimensions(split_dimension), split_count); in InferAllToAllShape()
2159 new_dimensions[split_dimension] /= split_count; in InferAllToAllShape()
2160 new_dimensions[concat_dimension] *= split_count; in InferAllToAllShape()
|
| H A D | shape_inference.h | 166 int64_t split_count);
|
| /aosp_15_r20/external/sdv/vsomeip/third_party/boost/intrusive/include/boost/intrusive/ |
| D | unordered_set.hpp | 373 size_type split_count() const;
854 size_type split_count() const;
|
| D | hashtable.hpp | 1469 BOOST_INTRUSIVE_FORCEINLINE SizeType split_count() const in split_count() function
1685 using internal_type::split_count;
3047 const size_type split_idx = this->split_count(); in incremental_rehash()
3087 size_type split_count() const;
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf_patterns.td | 314 …atcher ElementsAttr:$group_assignment), I64Attr:$concat_dimension, $split_dimension, $split_count),
315 …(HLO_AllToAllOp $input, $split_dimension, $concat_dimension, $split_count, (CastElementsToI64Eleme…
|
| /aosp_15_r20/external/libaom/av1/encoder/ |
| H A D | encodeframe_utils.h | 103 int split_count; member
|
| /aosp_15_r20/external/mesa3d/src/imagination/vulkan/ |
| H A D | pvr_dump_csb.c | 986 pvr_dump_field_member_u32(base_ctx, &index_list9, split_count); in print_block_vdmctrl_index_list()
988 pvr_dump_field_member_not_present(base_ctx, &index_list9, split_count); in print_block_vdmctrl_index_list()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/xla_extension/ |
| H A D | ops.pyi | 118 split_count: int,
|
| /aosp_15_r20/external/rust/android-crates-io/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/strings/internal/ |
| D | cord_rep_btree_test.cc | 1412 size_t split_count = 0; in TEST_P() local
1418 if (++split_count >= split_limit) { in TEST_P()
|