/external/tensorflow/tensorflow/python/ops/ |
D | collective_ops.py | 24 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,
|
D | collective_ops_test.py | 32 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.py | 349 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.py | 378 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/ |
D | collective_param_resolver_distributed.cc | 31 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 …]
|
D | collective_param_resolver_distributed.h | 51 bool GroupIsCached(int32 group_key) LOCKS_EXCLUDED(group_mu_);
|
D | collective_param_resolver_distributed_test.cc | 192 cp.group.group_key = kGroupKey; in DefineCollectiveParams()
|
/external/v8/tools/testrunner/testproc/ |
D | combiner.py | 45 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/ |
D | collective_ops.cc | 48 << " 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()
|
D | collective_nccl_reducer_test.cc | 117 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()
|
D | set_kernels.cc | 277 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/ |
D | collective_param_resolver_local.cc | 65 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()
|
D | ring_gatherer_test.cc | 177 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()
|
D | ring_reducer_test.cc | 199 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()
|
D | hierarchical_tree_broadcaster_test.cc | 258 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()
|
D | collective_param_resolver_local_test.cc | 163 cp->group.group_key = 1; in TEST_F() 206 cp->group.group_key = 1; in InitializeCollectiveParamsForBroadcast()
|
/external/tensorflow/tensorflow/python/distribute/ |
D | collective_all_reduce_strategy.py | 257 group_key = self._collective_keys.get_group_key([devices[0]]) 285 group_size, group_key, collective_instance_key) 291 group_key, collective_instance_key)
|
D | cross_device_utils.py | 356 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/ |
D | collective.h | 53 int32 group_key; member 59 : group_key(0), group_size(0), device_type(DEVICE_CPU), num_tasks(0) {} in CollGroupParams()
|
D | collective.cc | 51 return strings::StrCat("CollGroupParams {group_key=", group_key, in ToString()
|
/external/tensorflow/tensorflow/core/protobuf/ |
D | worker.proto | 496 // 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
|
D | config.proto | 496 // same group_key value (in a distributed computation where tasks
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.raw_ops.pbtxt | 577 …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/ |
D | tensorflow.raw_ops.pbtxt | 577 …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/ |
D | wrappers.go | 4076 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…
|