Searched refs:group_assignment (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_optimizer.py | 38 group_assignment=None): argument
59 self._group_assignment = group_assignment
61 def _verify_and_get_subgroup_size(self, group_assignment, num_shards): argument
75 if not group_assignment:
77 if not (isinstance(group_assignment, list) and
78 all(isinstance(i, list) for i in group_assignment)):
80 group_assignment))
83 for g in group_assignment:
90 num_shards, group_assignment))
92 subgroup_size_list = [len(group) for group in group_assignment]
|
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 40 group_assignment = [list(range(num_shards))]
41 return group_assignment
48 group_assignment=None, argument
66 if group_assignment is None:
67 group_assignment = _create_default_group_assignment()
70 group_assignment,
92 def cross_replica_sum(x, group_assignment=None, name=None): argument
105 if group_assignment is None:
106 group_assignment = _create_default_group_assignment()
108 return gen_tpu_ops.cross_replica_sum(x, group_assignment, name=name)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_AllToAll.pbtxt | 11 name: "group_assignment"
14 [num_groups, num_replicas_per_group]. `group_assignment[i]` represents the
46 size(group_assignment.get_shape()[1])
52 `split_dimension` and send to the other replicas given group_assignment. After
60 group_assignment=`[[0, 1]]`
|
| D | api_def_CrossReplicaSum.pbtxt | 11 name: "group_assignment"
14 [num_groups, num_replicas_per_group]. `group_assignment[i]` represents the
35 Passing group_assignment=`[[0,2,4,6],[1,3,5,7]]` sets `A, C, E, G` as group 0,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 85 …argspec: "args=[\'input\', \'group_assignment\', \'concat_dimension\', \'split_dimension\', \'spli…
729 …argspec: "args=[\'input\', \'group_assignment\', \'name\'], varargs=None, keywords=None, defaults=…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 85 …argspec: "args=[\'input\', \'group_assignment\', \'concat_dimension\', \'split_dimension\', \'spli…
729 …argspec: "args=[\'input\', \'group_assignment\', \'name\'], varargs=None, keywords=None, defaults=…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 20923 func CrossReplicaSum(scope *Scope, input tf.Output, group_assignment tf.Output) (output tf.Output) {
20930 input, group_assignment,
25654 func AllToAll(scope *Scope, input tf.Output, group_assignment tf.Output, concat_dimension int64, sp…
25662 input, group_assignment,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 1556 name: "group_assignment"
1593 name: "group_assignment"
16639 name: "group_assignment"
16664 name: "group_assignment"
|
| D | ops_history.v2.pbtxt | 1556 name: "group_assignment"
1593 name: "group_assignment"
16581 name: "group_assignment"
16606 name: "group_assignment"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 661 name: "group_assignment"
7202 name: "group_assignment"
|