Home
last modified time | relevance | path

Searched refs:group_key (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/python/ops/
Dcollective_ops.py24 def all_reduce(t, group_size, group_key, instance_key, merge_op, final_op, argument
54 group_key=group_key,
61 def all_gather(t, group_size, group_key, instance_key): argument
86 group_key=group_key,
90 def broadcast_send(t, shape, dtype, group_size, group_key, instance_key): argument
133 group_key=group_key,
137 def broadcast_recv(shape, dtype, group_size, group_key, instance_key): argument
161 group_key=group_key,
Dcollective_ops_test.py32 group_key = 1
38 colred0 = collective_ops.all_reduce(in0, 2, group_key, instance_key,
42 colred1 = collective_ops.all_reduce(in1, 2, group_key, instance_key,
52 group_key = 1
64 in_tensor, group_size, group_key, instance, 'Add', 'Div'))
93 group_key = 1
100 2, group_key, instance_key)
104 2, group_key, instance_key)
116 group_key = 1
122 colred0 = collective_ops.all_gather(in0, 2, group_key, instance_key)
[all …]
/external/python/cpython2/Lib/
D_strptime.py349 for group_key in found_dict.iterkeys():
355 if group_key == 'y':
364 elif group_key == 'Y':
366 elif group_key == 'm':
368 elif group_key == 'B':
370 elif group_key == 'b':
372 elif group_key == 'd':
374 elif group_key == 'H':
376 elif group_key == 'I':
392 elif group_key == 'M':
[all …]
/external/python/cpython3/Lib/
D_strptime.py378 for group_key in found_dict.keys():
384 if group_key == 'y':
393 elif group_key == 'Y':
395 elif group_key == 'G':
397 elif group_key == 'm':
399 elif group_key == 'B':
401 elif group_key == 'b':
403 elif group_key == 'd':
405 elif group_key == 'H':
407 elif group_key == 'I':
[all …]
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed.cc31 req_.set_group_key(group.group_key); in CompleteGroupCall()
58 req_.set_group_key(group.group_key); in CompleteInstanceCall()
116 cp.group.group_key = request->group_key(); in CompleteGroupAsync()
127 response->set_group_key(gr->group.group_key); in CompleteGroupAsync()
149 cp->group.group_key = request->group_key(); in CompleteInstanceAsync()
209 bool CollectiveParamResolverDistributed::GroupIsCached(int32 group_key) { in GroupIsCached() argument
211 const auto& it = group_table_.find(group_key); in GroupIsCached()
221 gr->group.group_key = resp.group_key(); in UpdateGroupCache()
245 auto it = group_table_.find(gr->group.group_key); in UpdateGroupCache()
247 group_table_[gr->group.group_key] = std::move(gr); in UpdateGroupCache()
[all …]
Dcollective_param_resolver_distributed.h51 bool GroupIsCached(int32 group_key) LOCKS_EXCLUDED(group_mu_);
Dcollective_param_resolver_distributed_test.cc192 cp.group.group_key = kGroupKey; in DefineCollectiveParams()
/external/v8/tools/testrunner/testproc/
Dcombiner.py45 group_key = self._get_group_key(test)
46 if not group_key:
50 self._groups[test.suite.name].add_test(group_key, test)
122 group_key = rng.choice(self._keys)
123 tests = self._groups[group_key]
/external/tensorflow/tensorflow/core/kernels/
Dcollective_ops.cc48 << " group " << col_params_.group.group_key << " instance " in CanProceedWithCompute()
77 OP_REQUIRES_OK(c, c->GetAttr("group_key", &col_params_.group.group_key)); in CollectiveGatherOpKernel()
128 OP_REQUIRES_OK(c, c->GetAttr("group_key", &col_params_.group.group_key)); in CollectiveReduceOpKernel()
217 << " group " << col_params_.group.group_key << " instance " in ComputeAsync()
237 OP_REQUIRES_OK(c, c->GetAttr("group_key", &col_params_.group.group_key)); in CollectiveBcastSendOpKernel()
286 << " group " << col_params_.group.group_key << " instance " in ComputeAsync()
306 OP_REQUIRES_OK(c, c->GetAttr("group_key", &col_params_.group.group_key)); in CollectiveBcastRecvOpKernel()
348 << " group " << col_params_.group.group_key << " instance " in ComputeAsync()
Dcollective_nccl_reducer_test.cc117 const int group_key = 5; in Init() local
118 col_params_.group.group_key = group_key; in Init()
207 .Attr("group_key", params.group.group_key) in GetCollectiveReduce()
224 col_params_.group.group_key = parent_->col_params_.group.group_key; in DeviceInstance()
Dset_kernels.cc277 const auto group_key = group.group(); in Compute() local
279 group_key.begin(), group_key.end(), output_strides.begin(), 0LL); in Compute()
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.cc65 auto it = group_table_.find(cp->group.group_key); in CompleteGroupLocal()
68 gr->group.group_key = cp->group.group_key; in CompleteGroupLocal()
71 group_table_[gr->group.group_key].reset(gr); in CompleteGroupLocal()
72 VLOG(2) << "New group_key=" << gr->group.group_key in CompleteGroupLocal()
87 " and group_key ", cp->group.group_key, " but that group has type ", in CompleteGroupLocal()
92 cp->group.group_size, " and group_key", cp->group.group_key, in CompleteGroupLocal()
104 " and group_key ", cp->group.group_key, in CompleteGroupLocal()
118 VLOG(1) << "group_key=" << gr->group.group_key in CompleteGroupLocal()
638 DCHECK_EQ(cp->group.group_key, gr->group.group_key); in CompleteInstanceLocal()
Dring_gatherer_test.cc177 col_params_.group.group_key = kGroupKey; in Init()
353 .Attr("group_key", params.group.group_key) in GetCollectiveGather()
388 col_params_.group.group_key = parent_->col_params_.group.group_key; in DeviceInstance()
533 cp.group.group_key = 1; in SetUpCollectiveParams()
Dring_reducer_test.cc199 col_params_.group.group_key = kGroupKey; in Init()
376 .Attr("group_key", params.group.group_key) in GetCollectiveReduce()
412 col_params_.group.group_key = parent_->col_params_.group.group_key; in DeviceInstance()
560 cp.group.group_key = 1; in SetUpCollectiveParams()
Dhierarchical_tree_broadcaster_test.cc258 col_params_.group.group_key = kGroupKey; in Init()
381 .Attr("group_key", params.group.group_key) in GetCollectiveBcastSend()
399 .Attr("group_key", params.group.group_key) in GetCollectiveBcastRecv()
563 col_params_.group.group_key = parent_->col_params_.group.group_key; in DeviceInstance()
Dcollective_param_resolver_local_test.cc163 cp->group.group_key = 1; in TEST_F()
206 cp->group.group_key = 1; in InitializeCollectiveParamsForBroadcast()
/external/tensorflow/tensorflow/python/distribute/
Dcollective_all_reduce_strategy.py257 group_key = self._collective_keys.get_group_key([devices[0]])
285 group_size, group_key, collective_instance_key)
291 group_key, collective_instance_key)
Dcross_device_utils.py356 group_key = collective_keys.get_group_key(devices)
367 input_tensors[d], group_size, group_key, instance_key, reduction_op,
/external/tensorflow/tensorflow/core/framework/
Dcollective.h53 int32 group_key; member
59 : group_key(0), group_size(0), device_type(DEVICE_CPU), num_tasks(0) {} in CollGroupParams()
Dcollective.cc51 return strings::StrCat("CollGroupParams {group_key=", group_key, in ToString()
/external/tensorflow/tensorflow/core/protobuf/
Dworker.proto496 // group_key. Service will respond when all group_size devices become known.
499 int32 group_key = 1; field
505 // Gives the complete membership of the group identified by group_key.
507 int32 group_key = 1; field
524 int32 group_key = 5; field
Dconfig.proto496 // same group_key value (in a distributed computation where tasks
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt577 …argspec: "args=[\'T\', \'group_size\', \'group_key\', \'instance_key\', \'shape\', \'name\'], vara…
581 …argspec: "args=[\'input\', \'group_size\', \'group_key\', \'instance_key\', \'shape\', \'name\'], …
585 …argspec: "args=[\'input\', \'group_size\', \'group_key\', \'instance_key\', \'shape\', \'name\'], …
593 …argspec: "args=[\'input\', \'group_size\', \'group_key\', \'instance_key\', \'merge_op\', \'final_…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt577 …argspec: "args=[\'T\', \'group_size\', \'group_key\', \'instance_key\', \'shape\', \'name\'], vara…
581 …argspec: "args=[\'input\', \'group_size\', \'group_key\', \'instance_key\', \'shape\', \'name\'], …
585 …argspec: "args=[\'input\', \'group_size\', \'group_key\', \'instance_key\', \'shape\', \'name\'], …
593 …argspec: "args=[\'input\', \'group_size\', \'group_key\', \'instance_key\', \'merge_op\', \'final_…
/external/tensorflow/tensorflow/go/op/
Dwrappers.go4076 func CollectiveBcastSend(scope *Scope, input tf.Output, group_size int64, group_key int64, instance…
4080 …attrs := map[string]interface{}{"group_size": group_size, "group_key": group_key, "instance_key": …
4093 func CollectiveGather(scope *Scope, input tf.Output, group_size int64, group_key int64, instance_ke…
4097 …attrs := map[string]interface{}{"group_size": group_size, "group_key": group_key, "instance_key": …
4121 func CollectiveReduce(scope *Scope, input tf.Output, group_size int64, group_key int64, instance_ke…
4125 …attrs := map[string]interface{}{"group_size": group_size, "group_key": group_key, "instance_key": …
27191 func CollectiveBcastRecv(scope *Scope, T tf.DataType, group_size int64, group_key int64, instance_k…
27195 …attrs := map[string]interface{}{"T": T, "group_size": group_size, "group_key": group_key, "instanc…

12