| /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 | 47 split_count, argument
73 split_count=split_count,
88 split_count=op.get_attr("split_count")), None
|
| /external/v8/gni/ |
| D | v8.gni | 133 if (defined(invoker.split_count) && invoker.split_count > 1 &&
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1233 split_count = get_replica_count()
1235 split_count = len(replica_groups[0])
1236 if not all(split_count == len(g) for g in replica_groups):
1239 split_count, replica_groups_protos)
|
| D | local_computation_builder.cc | 500 int64 split_count, absl::Span<const ReplicaGroup> replica_groups) { in AllToAll() argument
507 split_count, rg); in AllToAll()
|
| D | local_computation_builder.h | 256 int64 concat_dimension, int64 split_count,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 136 int64 split_count);
|
| D | shape_inference.cc | 1993 int64 split_count) { in InferAllToAllShape() argument
1994 TF_RET_CHECK(split_count > 0); in InferAllToAllShape()
2005 if (shape.dimensions(split_dimension) % split_count != 0) { in InferAllToAllShape()
2009 shape.dimensions(split_dimension), split_count); in InferAllToAllShape()
2013 new_dimensions[split_dimension] /= split_count; in InferAllToAllShape()
2014 new_dimensions[concat_dimension] *= split_count; in InferAllToAllShape()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2160 int64 concat_dimension, int64 split_count, in AllToAll() argument
2174 concat_dimension, split_count) in AllToAll()
2179 slices.reserve(split_count); in AllToAll()
2181 operand_shape.dimensions(split_dimension) / split_count; in AllToAll()
2182 for (int i = 0; i < split_count; i++) { in AllToAll()
2206 received.reserve(split_count); in AllToAll()
2207 for (int i = 0; i < split_count; i++) { in AllToAll()
3245 int64 concat_dimension, int64 split_count, in AllToAll() argument
3248 split_count, replica_groups); in AllToAll()
|
| D | xla_builder.h | 475 int64 concat_dimension, int64 split_count,
891 int64 concat_dimension, int64 split_count,
1594 int64 concat_dimension, int64 split_count,
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 40 1. the scatter phase. On each core, the operand is split into `split_count`
59 split_count.
62 <b> `AllToAll(operand, split_dimension, concat_dimension, split_count,
77 | `split_count` | `int64` | the number of cores that |
92 AllToAll(x, /*split_dimension=*/1, /*concat_dimension=*/0, /*split_count=*/4);
|
| /external/v8/ |
| D | BUILD.gn | 1499 split_count = 2
|
| /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 | 25654 …p_assignment tf.Output, concat_dimension int64, split_dimension int64, split_count int64) (output …
25658 …oncat_dimension": concat_dimension, "split_dimension": split_dimension, "split_count": split_count}
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 1582 name: "split_count"
1635 name: "split_count"
|
| D | ops_history.v1.pbtxt | 1582 name: "split_count"
1635 name: "split_count"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 703 name: "split_count"
|