Searched defs:split_count (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | tpu_cross_replica_ops.cc | 45 int split_count; in __anond991e9790102() local
|
| /external/rust/crates/rayon/src/iter/ |
| D | par_bridge.rs | 93 split_count: AtomicUsize, field
|
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 43 split_count, argument
|
| /external/libaom/av1/encoder/ |
| D | encodeframe_utils.h | 103 int split_count; member
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 1027 uint64_t split_count = all_to_all->replica_groups()[0].replica_ids_size(); in ImportInstructionImpl() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 3078 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/angle/third_party/abseil-cpp/absl/strings/internal/ |
| D | cord_rep_btree_test.cc | 1412 size_t split_count = 0; in TEST_P() local
|
| /external/webrtc/third_party/abseil-cpp/absl/strings/internal/ |
| D | cord_rep_btree_test.cc | 1412 size_t split_count = 0; in TEST_P() local
|
| /external/cronet/third_party/abseil-cpp/absl/strings/internal/ |
| D | cord_rep_btree_test.cc | 1412 size_t split_count = 0; in TEST_P() local
|
| /external/abseil-cpp/absl/strings/internal/ |
| D | cord_rep_btree_test.cc | 1409 size_t split_count = 0; in TEST_P() local
|
| /external/tensorflow/third_party/absl/abseil-cpp/absl/strings/internal/ |
| D | cord_rep_btree_test.cc | 1409 size_t split_count = 0; in TEST_P() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 2139 int64_t split_count) { in InferAllToAllShape()
|
| D | hlo_verifier.cc | 497 int64_t split_count = GetSubgroupSize(all_to_all, group_mode); in HandleAllToAll() local
|