/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()
|