| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 25 group_key, argument
65 group_key=group_key,
76 group_key, argument
114 group_key=group_key,
125 group_key, argument
156 group_key=group_key,
164 group_key, argument
196 group_key=group_key,
207 group_key, argument
258 group_key=group_key,
[all …]
|
| D | collective_ops_gpu_test.py | 64 group_key = 1
74 t, self._group_size, group_key, instance_key, 'Add', 'Div'))
84 group_key = 1
93 t, self._group_size, group_key, instance_key, 'Add', 'Div')
106 group_key = 1
116 t, self._group_size, group_key, instance_key, 'Add', 'Div'))
128 group_key = 1
138 t, self._group_size, group_key, instance_key, 'Add', 'Div',
149 group_key = 1
158 t, t.shape, t.dtype, self._group_size, group_key, instance_key))
[all …]
|
| D | collective_ops_test.py | 60 group_key = 1
78 group_key,
94 group_key = 1
106 in_tensor, group_size, group_key, instance, 'Add', 'Div'))
184 group_key = 1
217 input_tensor, group_size, group_key, instances[j],
245 group_key = 1
267 colred0 = collective_ops.all_reduce(input0, group_size, group_key,
269 colred1 = collective_ops.all_reduce(input1, group_size, group_key,
304 group_key = 1
[all …]
|
| D | collective_ops_xla_test.py | 35 group_key = 1
66 input_tensor1, group_size, group_key, instance_key1, 'Add', 'Id')
68 input_tensor2, group_size, group_key, instance_key2, 'Add', 'Id')
|
| D | collective_ops_benchmark.py | 43 group_key = collective_key_counter
56 r = collective_ops.all_reduce(t, group_size, group_key,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_test.py | 56 def all_reduce(t, group_size, group_key, instance_key, *args, **kwargs): argument
58 group_key = array_ops.identity(group_key)
60 return _collective_ops.all_reduce_v2(t, group_size, group_key, instance_key,
64 def all_gather(t, group_size, group_key, instance_key, *args, **kwargs): argument
66 group_key = array_ops.identity(group_key)
68 return _collective_ops.all_gather_v2(t, group_size, group_key, instance_key,
72 def broadcast_send(t, shape, dtype, group_size, group_key, instance_key, argument
75 group_key = array_ops.identity(group_key)
77 return _collective_ops.broadcast_send_v2(t, group_size, group_key,
81 def broadcast_recv(shape, dtype, group_size, group_key, instance_key, *args, argument
[all …]
|
| D | collective_ops_multi_worker_test.py | 116 group_key=100,
179 group_key = 100
189 group_key,
207 group_key,
217 group_key,
229 group_key,
240 group_key = 100
257 collective_ops.all_reduce(in_tensor, group_size, group_key,
263 collective_ops.all_reduce(in_tensor, group_size, group_key,
271 collective_ops.all_reduce(in_tensor, group_size, group_key, instance_key)
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_ops.cc | 28 static string CollectiveKey(OpKernelContext* ctx, int32 group_key, in CollectiveKey() argument
30 return strings::StrCat(group_key, ":", instance_key, ":", in CollectiveKey()
93 return CollectiveKey(c, col_params_->group.group_key, in GetCollectiveKey()
109 << " group " << col_params_->group.group_key << " instance " in CanProceedWithCompute()
148 OP_REQUIRES_OK(c, c->GetAttr("group_key", &col_params_->group.group_key)); in CollectiveGatherOpKernel()
186 << " group " << col_params->group.group_key << " instance " in ComputeAsyncImpl()
194 << " group " << col_params_->group.group_key << " instance " in ComputeAsyncImpl()
219 OP_REQUIRES_OK(c, c->GetAttr("group_key", &col_params_->group.group_key)); in CollectiveReduceOpKernel()
294 << " group " << col_params->group.group_key << " instance " in ComputeAsyncImpl()
302 << " group " << col_params_->group.group_key << " instance " in ComputeAsyncImpl()
[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 | 369 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/ |
| D | collective_param_resolver_distributed.cc | 40 req_.set_group_key(group.group_key); in CompleteGroupCall()
68 req_.set_group_key(group.group_key); in CompleteInstanceCall()
142 cp->group.group_key = request->group_key(); in CompleteGroupAsync()
151 response->set_group_key(gr->group.group_key); in CompleteGroupAsync()
170 GroupRec* gr = GetCachedGroup(request->group_key()); in CompleteInstanceAsync()
173 "group ", request->group_key(), in CompleteInstanceAsync()
181 "group ", request->group_key(), in CompleteInstanceAsync()
188 cp->group.group_key = request->group_key(); in CompleteInstanceAsync()
223 CollectiveParamResolverDistributed::GetCachedGroup(int32 group_key) { in GetCachedGroup() argument
225 auto it = group_table_.find(group_key); in GetCachedGroup()
[all …]
|
| D | collective_param_resolver_distributed.h | 56 GroupRec* GetCachedGroup(int32 group_key) TF_LOCKS_EXCLUDED(group_mu_);
74 bool InstanceIsCached(int32 group_key, int32 instance_key)
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_param_resolver_local_test.cc | 170 cp->group.group_key = 1; in TEST_F()
214 cp->group.group_key = 1; in InitializeCollectiveParamsForBroadcast()
303 CollectiveParams* MakeCollectiveParams(int group_key, int instance_key, in MakeCollectiveParams() argument
306 cp->group.group_key = group_key; in MakeCollectiveParams()
347 int group_key = 100; in TEST_F() local
353 Env::Default()->SchedClosure([this, group_key, instance_key, i, in TEST_F()
357 cp[i] = MakeCollectiveParams(group_key, instance_key, in TEST_F()
373 Env::Default()->SchedClosure([this, group_key, instance_key, i, &cancel_mgr, in TEST_F()
377 cp[i] = MakeCollectiveParams(group_key, instance_key + 1, in TEST_F()
396 int group_key = 100; in TEST_F() local
[all …]
|
| D | collective_param_resolver_local.cc | 136 auto it = group_table_.find(cp->group.group_key); in CompleteGroupLocal()
140 gr->group.group_key = cp->group.group_key; in CompleteGroupLocal()
169 group_table_[gr->group.group_key].reset(gr); in CompleteGroupLocal()
170 VLOG(2) << "New group_key=" << gr->group.group_key in CompleteGroupLocal()
199 " and group_key ", cp->group.group_key, " but that group has type ", in CompleteGroupLocal()
204 cp->group.group_size, " and group_key ", cp->group.group_key, in CompleteGroupLocal()
217 device.name(), " and group_key ", cp->group.group_key, in CompleteGroupLocal()
228 VLOG(1) << "CompleteGroupLocal group_key=" << gr->group.group_key in CompleteGroupLocal()
594 VLOG(2) << "Optimized device order for group " << gp->group_key << ": " in CompleteDefaultRanking()
607 auto group_it = instance_table_.find(gr->group.group_key); in GetOrCreateInstanceRec()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | CollectiveBcastRecv.pbtxt | 25 name: "group_key"
62 name: "group_key"
107 name: "group_key"
|
| D | CollectiveBcastSend.pbtxt | 29 name: "group_key"
70 name: "group_key"
119 name: "group_key"
|
| D | CollectiveReduce.pbtxt | 29 name: "group_key"
92 name: "group_key"
163 name: "group_key"
|
| D | CollectiveGather.pbtxt | 29 name: "group_key"
70 name: "group_key"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | CollectiveBcastSend.pbtxt | 29 name: "group_key"
71 name: "group_key"
120 name: "group_key"
|
| D | CollectiveBcastRecv.pbtxt | 25 name: "group_key"
63 name: "group_key"
108 name: "group_key"
|
| D | CollectiveGather.pbtxt | 29 name: "group_key"
70 name: "group_key"
118 name: "group_key"
|
| D | CollectiveReduce.pbtxt | 29 name: "group_key"
92 name: "group_key"
163 name: "group_key"
241 name: "group_key"
|
| D | CollectiveReduceV2.pbtxt | 12 name: "group_key"
78 name: "group_key"
151 name: "group_key"
|
| D | CollectiveGatherV2.pbtxt | 12 name: "group_key"
63 name: "group_key"
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 219 def get_instance_key(self, group_key, device): argument
238 group = self._instance_key_table.get(group_key, None)
240 raise ValueError('group {} not found'.format(group_key))
242 raise ValueError('{} not in group {}'.format(device, group_key))
264 group_key, argument
268 self._group_key = group_key
|