Searched defs:split_count (Results 1 – 8 of 8) sorted by relevance
| /external/rust/crates/rayon/src/iter/ |
| D | par_bridge.rs | 81 let split_count = AtomicUsize::new(current_num_threads()); in drive_unindexed() localVariable
100 split_count: &'a AtomicUsize, field
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | tpu_cross_replica_ops.cc | 43 int split_count; in __anonc85edbdb0102() local
|
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 48 split_count, argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2858 int64 concat_dimension, int64 split_count, in AllToAll()
2873 int64 split_count, const std::vector<ReplicaGroup>& replica_groups) { in AllToAllArray()
2924 int64 concat_dimension, int64 split_count, in AllToAllTuple()
4678 int64 concat_dimension, int64 split_count, in AllToAll()
4686 int64 concat_dimension, int64 split_count, in AllToAllTuple()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.cc | 307 const int64 split_count = hlo->replica_groups().empty() in HandleAllToAll() local
|
| D | shape_inference.cc | 2063 int64 split_count) { in InferAllToAllShape()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 683 auto split_count = op.split_count(); in Verify() local
|
| /external/libaom/libaom/av1/encoder/ |
| D | encodeframe.c | 176 int split_count; member
|