| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 26 instance_key, argument
66 instance_key=instance_key,
77 instance_key, argument
115 instance_key=instance_key,
126 instance_key, argument
157 instance_key=instance_key,
165 instance_key, argument
197 instance_key=instance_key,
208 instance_key, argument
259 instance_key=instance_key,
[all …]
|
| D | collective_ops_gpu_test.py | 65 instance_key = 1
74 t, self._group_size, group_key, instance_key, 'Add', 'Div'))
85 instance_key = 50
93 t, self._group_size, group_key, instance_key, 'Add', 'Div')
107 instance_key = 100
116 t, self._group_size, group_key, instance_key, 'Add', 'Div'))
129 instance_key = 1
138 t, self._group_size, group_key, instance_key, 'Add', 'Div',
150 instance_key = 1
158 t, t.shape, t.dtype, self._group_size, group_key, instance_key))
[all …]
|
| D | collective_ops_test.py | 55 instance_key=1, argument
79 instance_key,
288 instance_key=30,
299 instance_key=40,
305 instance_key = 1
311 2, group_key, instance_key)
315 2, group_key, instance_key)
334 instance_key = 1
339 c0 = collective_ops.all_gather(in0, 2, group_key, instance_key)
342 c1 = collective_ops.all_gather(in1, 2, group_key, instance_key)
[all …]
|
| D | collective_ops_benchmark.py | 44 instance_key = collective_key_counter
57 instance_key, "Add", "Div")
|
| /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
59 instance_key = array_ops.identity(instance_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
67 instance_key = array_ops.identity(instance_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
76 instance_key = array_ops.identity(instance_key)
78 instance_key, *args, **kwargs)
81 def broadcast_recv(shape, dtype, group_size, group_key, instance_key, *args, argument
[all …]
|
| D | collective_ops_multi_worker_test.py | 117 instance_key=100,
180 instance_key = 100
190 instance_key,
208 instance_key,
218 instance_key,
230 instance_key,
241 instance_key = 100
258 instance_key)
264 instance_key)
271 collective_ops.all_reduce(in_tensor, group_size, group_key, instance_key)
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_ops.cc | 29 int32 instance_key) { in CollectiveKey() argument
30 return strings::StrCat(group_key, ":", instance_key, ":", in CollectiveKey()
94 col_params_->instance.instance_key); in GetCollectiveKey()
110 << col_params_->instance.instance_key; in CanProceedWithCompute()
150 c, c->GetAttr("instance_key", &col_params_->instance.instance_key)); in CollectiveGatherOpKernel()
187 << col_params->instance.instance_key << " status " << s; in ComputeAsyncImpl()
195 << col_params_->instance.instance_key; in ComputeAsyncImpl()
221 c, c->GetAttr("instance_key", &col_params_->instance.instance_key)); in CollectiveReduceOpKernel()
247 << col_params_->instance.instance_key << " merge_op " in CollectiveReduceOpKernel()
295 << col_params->instance.instance_key << " status " << s; in ComputeAsyncImpl()
[all …]
|
| D | collective_nccl_test.cc | 110 void Init(const int num_ranks, const int instance_key) { in Init() argument
137 col_params_->instance.instance_key = instance_key; in Init()
185 void RunTest(int num_ranks, int input_length, int instance_key) { in RunTest() argument
191 Init(num_ranks, instance_key); in RunTest()
240 .Attr("instance_key", params.instance.instance_key) in GetCollectiveReduceOpKernel()
327 strings::StrCat(col_params_->instance.instance_key, ":0:0"); in RunReduce()
359 strings::StrCat(col_params_->instance.instance_key, ":0:0"); in RunBroadcast()
400 strings::StrCat(col_params_->instance.instance_key, ":0:0"); in RunGather()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_param_resolver_local_test.cc | 174 cp->instance.instance_key = 7; in TEST_F()
211 void InitializeCollectiveParamsForBroadcast(int instance_key, int device_idx, in InitializeCollectiveParamsForBroadcast() argument
218 cp->instance.instance_key = instance_key; in InitializeCollectiveParamsForBroadcast()
303 CollectiveParams* MakeCollectiveParams(int group_key, int instance_key, in MakeCollectiveParams() argument
310 cp->instance.instance_key = instance_key; in MakeCollectiveParams()
348 int instance_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()
[all …]
|
| D | graph_execution_state.cc | 911 int32 instance_key; in BuildGraph() local
913 GetNodeAttr(node->attrs(), "instance_key", &instance_key)); in BuildGraph()
914 instance_key_set.emplace(instance_key); in BuildGraph()
925 int32 instance_key; in BuildGraph() local
927 GetNodeAttr(ndef, "instance_key", &instance_key)); in BuildGraph()
928 instance_key_set.emplace(instance_key); in BuildGraph()
938 for (int32 instance_key : instance_key_set) { in BuildGraph() local
939 hash = Hash64Combine(instance_key, hash); in BuildGraph()
|
| D | base_collective_executor.cc | 458 if (launched_.find(col_params.instance.instance_key) == launched_.end()) { in UnblockDependencies()
463 launched_[col_params.instance.instance_key] = num_devices; in UnblockDependencies()
465 if (--launched_[col_params.instance.instance_key] == 0) { in UnblockDependencies()
467 << col_params.instance.instance_key; in UnblockDependencies()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | CollectiveBcastRecv.pbtxt | 29 name: "instance_key"
66 name: "instance_key"
111 name: "instance_key"
|
| D | CollectiveBcastSend.pbtxt | 33 name: "instance_key"
74 name: "instance_key"
123 name: "instance_key"
|
| D | CollectiveReduce.pbtxt | 33 name: "instance_key"
96 name: "instance_key"
167 name: "instance_key"
|
| D | CollectiveGather.pbtxt | 33 name: "instance_key"
74 name: "instance_key"
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed.cc | 70 req_.set_instance_key(instance.instance_key); in CompleteInstanceCall()
192 cp->instance.instance_key = request->instance_key(); in CompleteInstanceAsync()
213 response->set_instance_key(cp->instance.instance_key); in CompleteInstanceAsync()
331 int32 instance_key) { in InstanceIsCached() argument
337 auto instance_it = group_it->second.find(instance_key); in InstanceIsCached()
355 cp->instance.instance_key, " gives source_rank=", source_rank, in UpdateInstanceCache()
367 cp->instance.instance_key, " has known.size()=", ir->known.size(), in UpdateInstanceCache()
384 } else if (InstanceIsCached(cp->group.group_key, cp->instance.instance_key)) { in CompleteInstanceDistributed()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | CollectiveBcastRecv.pbtxt | 29 name: "instance_key"
67 name: "instance_key"
112 name: "instance_key"
|
| D | CollectiveBcastSend.pbtxt | 33 name: "instance_key"
75 name: "instance_key"
124 name: "instance_key"
|
| D | CollectiveGather.pbtxt | 33 name: "instance_key"
74 name: "instance_key"
122 name: "instance_key"
|
| D | CollectiveReduce.pbtxt | 33 name: "instance_key"
96 name: "instance_key"
167 name: "instance_key"
245 name: "instance_key"
|
| D | CollectiveReduceV2.pbtxt | 16 name: "instance_key"
82 name: "instance_key"
155 name: "instance_key"
|
| D | CollectiveGatherV2.pbtxt | 16 name: "instance_key"
67 name: "instance_key"
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | collective_order.cc | 39 int32 instance_key; in DiscoverDataDependencies() local
44 GetNodeAttr(node->attrs(), "instance_key", &instance_key); in DiscoverDataDependencies()
47 instance_keys->push_back(instance_key); in DiscoverDataDependencies()
58 child_deps.insert(instance_key); in DiscoverDataDependencies()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 318 instance_key = self._collective_keys.get_instance_key(
321 return ops.convert_to_tensor(instance_key, dtype=dtypes.int32)
353 instance_key = self._next_instance_key()
361 instance_key,
378 instance_key = self._next_instance_key()
385 instance_key,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 79 instance_key = other.instance_key; in operator =()
103 strings::StrCat("CollInstanceParams { instance_key=", instance_key, in ToString()
|