Home
last modified time | relevance | path

Searched defs:split_count (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dtpu_cross_replica_ops.cc45 int split_count; in __anon385994aa0102() local
/external/rust/android-crates-io/crates/rayon/src/iter/
Dpar_bridge.rs103 split_count: AtomicUsize, field
/external/sdv/vsomeip/third_party/boost/move/test/
Dbench_merge.cpp61 std::size_t split_count = L / 2; in generate_elements() local
/external/tensorflow/tensorflow/python/tpu/ops/
Dtpu_ops.py43 split_count, argument
/external/mesa3d/src/amd/vpelib/src/core/
Dresource.c380 int32_t split_count, split_idx; in calculate_recout() local
/external/libaom/av1/encoder/
Dencodeframe_utils.h103 int split_count; member
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc1027 uint64_t split_count = all_to_all->replica_groups()[0].replica_ids_size(); in ImportInstructionImpl() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc3078 int64_t concat_dimension, int64_t split_count, in AllToAll()
3092 int64_t concat_dimension, int64_t split_count, in AllToAllArray()
3184 int64_t concat_dimension, int64_t split_count, in AllToAllTuple()
4691 int64_t concat_dimension, int64_t split_count, in AllToAll()
4706 int64_t concat_dimension, int64_t split_count, in AllToAllTuple()
/external/pytorch/torch/csrc/jit/tensorexpr/
Dloopnest.cpp1573 ExprPtr split_count = alloc<Div>(size, factor_expr); in splitWithTail() local
1641 ExprPtr split_count = alloc<Div>( in splitWithMask() local
/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
/external/abseil-cpp/absl/strings/internal/
Dcord_rep_btree_test.cc1412 size_t split_count = 0; in TEST_P() local
/external/angle/third_party/abseil-cpp/absl/strings/internal/
Dcord_rep_btree_test.cc1412 size_t split_count = 0; in TEST_P() local
/external/cronet/stable/third_party/abseil-cpp/absl/strings/internal/
Dcord_rep_btree_test.cc1412 size_t split_count = 0; in TEST_P() local
/external/cronet/tot/third_party/abseil-cpp/absl/strings/internal/
Dcord_rep_btree_test.cc1412 size_t split_count = 0; in TEST_P() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc2139 int64_t split_count) { in InferAllToAllShape()
Dhlo_verifier.cc497 int64_t split_count = GetSubgroupSize(all_to_all, group_mode); in HandleAllToAll() local
/external/sdv/vsomeip/third_party/boost/intrusive/include/boost/intrusive/
Dhashtable.hpp1469 BOOST_INTRUSIVE_FORCEINLINE SizeType split_count() const in split_count() function