| /external/rust/crates/rayon/src/iter/ |
| D | par_bridge.rs | 81 let split_count = AtomicUsize::new(current_num_threads()); in drive_unindexed() localVariable
89 split_count: &split_count, in drive_unindexed()
100 split_count: &'a AtomicUsize, field
110 split_count: self.split_count, in clone()
125 let mut count = self.split_count.load(Ordering::SeqCst); in split()
132 self.split_count in split()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | tpu_cross_replica_ops.cc | 43 int split_count; in __anonc85edbdb0102() local
45 TF_RETURN_IF_ERROR(c->GetAttr("split_count", &split_count)); in __anonc85edbdb0102()
66 dims[i] = c->MakeDim(c->Value(dims[i]) * split_count); in __anonc85edbdb0102()
69 dims[i] = c->MakeDim(c->Value(dims[i]) / split_count); in __anonc85edbdb0102()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| 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
|
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 48 split_count, argument
74 split_count=split_count,
89 split_count=op.get_attr("split_count")), None
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | AllToAll.pbtxt | 34 name: "split_count"
87 name: "split_count"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | AllToAll.pbtxt | 34 name: "split_count"
87 name: "split_count"
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2858 int64 concat_dimension, int64 split_count, in AllToAll() argument
2865 split_count, replica_groups, layout); in AllToAll()
2867 return AllToAllArray(operand, split_dimension, concat_dimension, split_count, in AllToAll()
2873 int64 split_count, const std::vector<ReplicaGroup>& replica_groups) { in AllToAllArray() argument
2879 concat_dimension, split_count)); in AllToAllArray()
2884 for (int64 i = 0; i < split_count; ++i) { in AllToAllArray()
2905 sizes.push_back(split_count); in AllToAllArray()
2906 sizes.push_back(operand_shape->dimensions(i) / split_count); in AllToAllArray()
2924 int64 concat_dimension, int64 split_count, in AllToAllTuple() argument
2939 concat_dimension, split_count) in AllToAllTuple()
[all …]
|
| D | xla_builder.h | 747 int64 split_count,
752 int64 concat_dimension, int64 split_count,
1308 int64 concat_dimension, int64 split_count,
1312 int64 concat_dimension, int64 split_count,
1444 int64 concat_dimension, int64 split_count,
2224 int64 split_count,
2229 int64 concat_dimension, int64 split_count,
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy.py | 1328 split_count=self._num_replicas_in_sync)
1569 split_count=self.num_replicas_in_sync)
1582 split_count=self.num_replicas_in_sync)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 144 int64 split_count);
|
| D | shape_inference.cc | 2063 int64 split_count) { in InferAllToAllShape() argument
2064 TF_RET_CHECK(split_count > 0); in InferAllToAllShape()
2075 if (shape.dimensions(split_dimension) % split_count != 0) { in InferAllToAllShape()
2079 shape.dimensions(split_dimension), split_count); in InferAllToAllShape()
2083 new_dimensions[split_dimension] /= split_count; in InferAllToAllShape()
2084 new_dimensions[concat_dimension] *= split_count; in InferAllToAllShape()
|
| D | hlo_verifier.cc | 307 const int64 split_count = hlo->replica_groups().empty() in HandleAllToAll() local
311 if (g.replica_ids_size() != split_count) { in HandleAllToAll()
324 *all_to_all->split_dimension(), split_count)); in HandleAllToAll()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_patterns.td | 297 …atcher ElementsAttr:$group_assignment), I64Attr:$concat_dimension, $split_dimension, $split_count),
298 …(HLO_AllToAllOp $input, $split_dimension, $concat_dimension, $split_count, (CastElementsToI64Eleme…
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | ops.mlir | 21 split_count = 4 : i64,
34 split_count = 5 : i64,
43 // expected-error@+1 {{split dimension has size 16, expected to be a multiple of split_count 5}}
47 split_count = 5 : i64,
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 683 auto split_count = op.split_count(); in Verify() local
684 if (split_dim_size % split_count != 0) { in Verify()
687 << split_count; in Verify()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops_base.td | 806 - The scatter phase. On each core, the operand is split into `split_count`
826 `split_count`.
|
| D | hlo_ops.td | 606 I64Attr:$split_count,
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tensor_tracer.py | 1576 split_count=self._tt_config.num_replicas)
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 123 1. The scatter phase. On each core, the operand is split into `split_count`
145 `split_count`.
148 <b> `AllToAll(operand, split_dimension, concat_dimension, split_count,
163 | `split_count` | `int64` | the number of cores that |
178 AllToAll(x, /*split_dimension=*/1, /*concat_dimension=*/0, /*split_count=*/4);
|
| /external/libaom/libaom/av1/encoder/ |
| D | encodeframe.c | 176 int split_count; member
4520 sb_fp_stats->split_count = cpi->td.mb.txb_split_count; in backup_sb_state()
4553 cpi->td.mb.txb_split_count = sb_fp_stats->split_count; in restore_sb_state()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 234 On each replica, the input is split into `split_count` blocks along
236 receiving `split_count` - 1 blocks from other replicas, we concatenate the
246 split_count=2
260 I64Attr:$split_count
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 85 …\', \'group_assignment\', \'concat_dimension\', \'split_dimension\', \'split_count\', \'name\'], v…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 85 …\', \'group_assignment\', \'concat_dimension\', \'split_dimension\', \'split_count\', \'name\'], v…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 51322 …p_assignment tf.Output, concat_dimension int64, split_dimension int64, split_count int64) (output …
51326 …oncat_dimension": concat_dimension, "split_dimension": split_dimension, "split_count": split_count}
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 4884 …oAll"(%input, %group_assignment) {T = f32, concat_dimension = 1 : i64, split_count = 2 : i64, spli…
|