Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/ops/
Dcollective_ops.py23 def all_reduce(t, group_size, group_key, instance_key, merge_op, final_op, argument
55 group_key=group_key,
63 def all_gather(t, group_size, group_key, instance_key, argument
89 group_key=group_key,
94 def broadcast_send(t, shape, dtype, group_size, group_key, instance_key, argument
140 group_key=group_key,
145 def broadcast_recv(shape, dtype, group_size, group_key, instance_key, argument
174 group_key=group_key,
Dcollective_ops_test.py46 group_key = 1
59 tensor, group_size, group_key, instance_key, merge_op, final_op,
71 group_key = 1
83 in_tensor, group_size, group_key, instance, 'Add', 'Div'))
134 group_key = 1
167 input_tensor, group_size, group_key, instances[j],
194 group_key = 1
214 colred0 = collective_ops.all_reduce(input0, group_size, group_key,
216 colred1 = collective_ops.all_reduce(input1, group_size, group_key,
248 group_key = 1
[all …]
Dcollective_ops_gpu_test.py68 group_key = 1
80 t, self._group_size, group_key, instance_key, 'Add', 'Div'))
88 group_key = 1
100 t, self._group_size, group_key, instance_key, 'Add', 'Div'))
111 group_key = 1
123 t, self._group_size, group_key, instance_key, 'Add', 'Div'))
134 group_key = 1
147 t, self._group_size, group_key, instance_key, 'Add', 'Div',
156 group_key = 1
167 t, t.shape, t.dtype, self._group_size, group_key, instance_key))
[all …]
Dcollective_ops_benchmark.py43 group_key = collective_key_counter
56 r = collective_ops.all_reduce(t, group_size, group_key,
/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.py369 for group_key in found_dict.keys():
375 if group_key == 'y':
384 elif group_key == 'Y':
386 elif group_key == 'G':
388 elif group_key == 'm':
390 elif group_key == 'B':
392 elif group_key == 'b':
394 elif group_key == 'd':
396 elif group_key == 'H':
398 elif group_key == 'I':
[all …]
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed.cc33 req_.set_group_key(group.group_key); in CompleteGroupCall()
61 req_.set_group_key(group.group_key); in CompleteInstanceCall()
121 cp.group.group_key = request->group_key(); in CompleteGroupAsync()
133 response->set_group_key(gr->group.group_key); in CompleteGroupAsync()
157 cp->group.group_key = request->group_key(); in CompleteInstanceAsync()
213 bool CollectiveParamResolverDistributed::GroupIsCached(int32 group_key) { in GroupIsCached() argument
215 const auto& it = group_table_.find(group_key); in GroupIsCached()
225 gr->group.group_key = resp.group_key(); in UpdateGroupCache()
252 auto it = group_table_.find(gr->group.group_key); in UpdateGroupCache()
256 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_);
/external/tensorflow/tensorflow/core/kernels/
Dcollective_ops.cc48 << " group " << col_params_.group.group_key << " instance " in CanProceedWithCompute()
81 OP_REQUIRES_OK(c, c->GetAttr("group_key", &col_params_.group.group_key)); in CollectiveGatherOpKernel()
128 << " group " << col_params_.group.group_key << " instance " in ComputeAsync()
152 OP_REQUIRES_OK(c, c->GetAttr("group_key", &col_params_.group.group_key)); in CollectiveReduceOpKernel()
249 << " group " << col_params_.group.group_key << " instance " in ComputeAsync()
273 OP_REQUIRES_OK(c, c->GetAttr("group_key", &col_params_.group.group_key)); in CollectiveBcastSendOpKernel()
325 << " group " << col_params_.group.group_key << " instance " in ComputeAsync()
349 OP_REQUIRES_OK(c, c->GetAttr("group_key", &col_params_.group.group_key)); in CollectiveBcastRecvOpKernel()
394 << " group " << col_params_.group.group_key << " instance " in ComputeAsync()
Dcollective_nccl_test.cc125 const int group_key = num_ranks; in Init() local
126 col_params_.group.group_key = group_key; in Init()
237 .Attr("group_key", params.group.group_key) in GetCollectiveReduceOpKernel()
254 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/ops/compat/ops_history_v1/
DCollectiveBcastRecv.pbtxt25 name: "group_key"
62 name: "group_key"
107 name: "group_key"
DCollectiveBcastSend.pbtxt29 name: "group_key"
70 name: "group_key"
119 name: "group_key"
DCollectiveReduce.pbtxt29 name: "group_key"
92 name: "group_key"
163 name: "group_key"
DCollectiveGather.pbtxt29 name: "group_key"
70 name: "group_key"
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.cc86 auto it = group_table_.find(cp->group.group_key); in CompleteGroupLocal()
89 gr->group.group_key = cp->group.group_key; in CompleteGroupLocal()
117 group_table_[gr->group.group_key].reset(gr); in CompleteGroupLocal()
118 VLOG(2) << "New group_key=" << gr->group.group_key in CompleteGroupLocal()
139 " and group_key ", cp->group.group_key, " but that group has type ", in CompleteGroupLocal()
144 cp->group.group_size, " and group_key ", cp->group.group_key, in CompleteGroupLocal()
156 " and group_key ", cp->group.group_key, in CompleteGroupLocal()
175 VLOG(1) << "CompleteGroupLocal group_key=" << gr->group.group_key in CompleteGroupLocal()
710 DCHECK_EQ(cp->group.group_key, gr->group.group_key); in CompleteInstanceLocal()
Dring_gatherer_test.cc183 col_params_.group.group_key = kGroupKey; in Init()
360 .Attr("group_key", params.group.group_key) in GetCollectiveGather()
395 col_params_.group.group_key = parent_->col_params_.group.group_key; in DeviceInstance()
538 cp.group.group_key = 1; in SetUpCollectiveParams()
Dring_reducer_test.cc205 col_params_.group.group_key = kGroupKey; in Init()
384 .Attr("group_key", params.group.group_key) in GetCollectiveReduce()
420 col_params_.group.group_key = parent_->col_params_.group.group_key; in DeviceInstance()
566 cp.group.group_key = 1; in SetUpCollectiveParams()
Dhierarchical_tree_broadcaster_test.cc264 col_params_.group.group_key = kGroupKey; in Init()
387 .Attr("group_key", params.group.group_key) in GetCollectiveBcastSend()
405 .Attr("group_key", params.group.group_key) in GetCollectiveBcastRecv()
571 col_params_.group.group_key = parent_->col_params_.group.group_key; in DeviceInstance()
Dcollective_param_resolver_local_test.cc164 cp->group.group_key = 1; in TEST_F()
207 cp->group.group_key = 1; in InitializeCollectiveParamsForBroadcast()
/external/tensorflow/tensorflow/python/distribute/
Dcross_device_utils.py354 group_key = collective_keys.get_group_key(devices)
367 input_tensors[dev_idx], group_size, group_key, instance_key,
371 input_tensors[dev_idx], group_size, group_key, instance_key,
399 group_key = collective_keys.get_group_key(devices)
409 group_key, instance_key)
Dcollective_all_reduce_strategy.py360 group_key = self._collective_keys.get_group_key([device])
377 group_size, group_key, collective_instance_key)
383 group_size, group_key,
/external/tensorflow/tensorflow/core/framework/
Dcollective.h61 int32 group_key; member
68 : group_key(0), group_size(0), device_type(DEVICE_CPU), num_tasks(0) {} in CollGroupParams()
/external/tensorflow/tensorflow/core/protobuf/
Dworker.proto531 // group_key. Service will respond when all group_size devices become known.
534 int32 group_key = 1; field
541 // Gives the complete membership of the group identified by group_key.
543 int32 group_key = 1; field
561 int32 group_key = 5; field
/external/wpa_supplicant_8/src/common/
Ddpp_i.h82 EVP_PKEY * dpp_set_pubkey_point(EVP_PKEY *group_key, const u8 *buf, size_t len);

12