Home
last modified time | relevance | path

Searched refs:split_count (Results 1 – 25 of 52) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dtpu_cross_replica_ops.cc45 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/
Dunordered_test.hpp455 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 Dtfsa-2021-176.md19 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/
Dpar_bridge.rs93 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 Dapi_def_AllToAll.pbtxt43 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 Dtpu_ops.py43 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 Dtpu_test.py180 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 DAllToAll.pbtxt34 name: "split_count"
87 name: "split_count"
140 name: "split_count"
/aosp_15_r20/external/mesa3d/src/amd/vpelib/src/core/
H A Dresource.c326 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/
Dbench_merge.cpp61 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 DTableSplitter.cpp181 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 DAllToAll.pbtxt34 name: "split_count"
87 name: "split_count"
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.cc3078 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 Dxla_builder.h770 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 Dloopnest.cpp1573 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 Dtpu_strategy.py1453 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 Dshape_inference.cc2139 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 Dshape_inference.h166 int64_t split_count);
/aosp_15_r20/external/sdv/vsomeip/third_party/boost/intrusive/include/boost/intrusive/
Dunordered_set.hpp373 size_type split_count() const;
854 size_type split_count() const;
Dhashtable.hpp1469 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 Dlegalize_tf_patterns.td314 …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 Dencodeframe_utils.h103 int split_count; member
/aosp_15_r20/external/mesa3d/src/imagination/vulkan/
H A Dpvr_dump_csb.c986 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 Dops.pyi118 split_count: int,
/aosp_15_r20/external/rust/android-crates-io/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/strings/internal/
Dcord_rep_btree_test.cc1412 size_t split_count = 0; in TEST_P() local
1418 if (++split_count >= split_limit) { in TEST_P()

123