Home
last modified time | relevance | path

Searched refs:instance_key (Results 1 – 25 of 28) 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
55 instance_key=instance_key,
61 def all_gather(t, group_size, group_key, instance_key): argument
87 instance_key=instance_key)
90 def broadcast_send(t, shape, dtype, group_size, group_key, instance_key): argument
134 instance_key=instance_key)
137 def broadcast_recv(shape, dtype, group_size, group_key, instance_key): argument
162 instance_key=instance_key)
Dcollective_ops_test.py33 instance_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,
94 instance_key = 1
100 2, group_key, instance_key)
104 2, group_key, instance_key)
117 instance_key = 1
122 colred0 = collective_ops.all_gather(in0, 2, group_key, instance_key)
125 colred1 = collective_ops.all_gather(in1, 2, group_key, instance_key)
/external/tensorflow/tensorflow/core/kernels/
Dcollective_ops.cc31 return strings::StrCat(col_params_.instance.instance_key, ":", in GetCollectiveKey()
49 << col_params_.instance.instance_key; in CanProceedWithCompute()
79 c, c->GetAttr("instance_key", &col_params_.instance.instance_key)); in CollectiveGatherOpKernel()
130 c, c->GetAttr("instance_key", &col_params_.instance.instance_key)); in CollectiveReduceOpKernel()
208 int32 instance_key = col_params_.instance.instance_key; in ComputeAsync() local
209 auto actual_done = [c, instance_key, done](const Status& s) { in ComputeAsync()
213 << c->device()->name() << " instance " << instance_key; in ComputeAsync()
218 << instance_key; in ComputeAsync()
239 c, c->GetAttr("instance_key", &col_params_.instance.instance_key)); in CollectiveBcastSendOpKernel()
277 int32 instance_key = col_params_.instance.instance_key; in ComputeAsync() local
[all …]
Dcollective_nccl_reducer_test.cc121 const int instance_key = 23; in Init() local
122 col_params_.instance.instance_key = instance_key; in Init()
208 .Attr("instance_key", params.instance.instance_key) in GetCollectiveReduce()
289 strings::StrCat(col_params_.instance.instance_key, ":0:0"); in DoReduce()
Dcollective_nccl_reducer.cc154 << col_params_->instance.instance_key; in Run()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed.cc60 req_.set_instance_key(instance.instance_key); in CompleteInstanceCall()
153 cp->instance.instance_key = request->instance_key(); in CompleteInstanceAsync()
188 response->set_instance_key(cp->instance.instance_key); in CompleteInstanceAsync()
284 bool CollectiveParamResolverDistributed::InstanceIsCached(int32 instance_key) { in InstanceIsCached() argument
286 const auto& it = instance_table_.find(instance_key); in InstanceIsCached()
314 cp->instance.instance_key, " gives source_rank=", source_rank, in UpdateInstanceCache()
325 cp->instance.instance_key, in UpdateInstanceCache()
339 cp->instance.instance_key, " has known.size()=", ir->known.size(), in UpdateInstanceCache()
368 } else if (InstanceIsCached(cp->instance.instance_key)) { in CompleteInstanceDistributed()
Dcollective_param_resolver_distributed.h68 bool InstanceIsCached(int32 instance_key) LOCKS_EXCLUDED(instance_mu_);
Dcollective_param_resolver_distributed_test.cc196 cp.instance.instance_key = kInstanceKey; in DefineCollectiveParams()
/external/tensorflow/tensorflow/core/graph/
Dcollective_order.cc39 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()
Dcollective_order_test.cc81 int instance_key) { in CollectiveReduceNode() argument
90 .WithAttr("instance_key", instance_key) in CollectiveReduceNode()
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local_test.cc90 cp.instance.instance_key = 5; in TEST_F()
167 cp->instance.instance_key = 7; in TEST_F()
203 void InitializeCollectiveParamsForBroadcast(int instance_key, int device_idx, in InitializeCollectiveParamsForBroadcast() argument
210 cp->instance.instance_key = instance_key; in InitializeCollectiveParamsForBroadcast()
Dbase_collective_executor.cc335 if (launched_.find(col_params.instance.instance_key) == launched_.end()) { in Launched()
340 launched_[col_params.instance.instance_key] = num_devices; in Launched()
342 if (--launched_[col_params.instance.instance_key] == 0) { in Launched()
344 << col_params.instance.instance_key; in Launched()
Dgraph_execution_state.cc799 int32 instance_key; in BuildGraph() local
801 GetNodeAttr(node->attrs(), "instance_key", &instance_key)); in BuildGraph()
802 instance_key_set.emplace(instance_key); in BuildGraph()
810 int32 instance_key; in BuildGraph() local
812 GetNodeAttr(ndef, "instance_key", &instance_key)); in BuildGraph()
813 instance_key_set.emplace(instance_key); in BuildGraph()
821 for (int32 instance_key : instance_key_set) { in BuildGraph() local
822 hash = Hash64Combine(instance_key, hash); in BuildGraph()
Dcollective_param_resolver_local.cc507 auto it = instance_table_.find(cp->instance.instance_key); in FindInstanceRec()
524 instance_table_[cp->instance.instance_key].reset(irec); in FindInstanceRec()
634 << " instance_key: " << cp->instance.instance_key << " gr " << gr; in CompleteInstanceLocal()
729 ir->status = errors::Internal("Instance ", cp->instance.instance_key, in WaitForGroup()
741 errors::Internal("Instance ", cp->instance.instance_key, in WaitForGroup()
761 errors::Internal("Instance ", cp->instance.instance_key, in WaitForGroup()
768 "Instance ", cp->instance.instance_key, " device ", in WaitForGroup()
Dhierarchical_tree_broadcaster_test.cc260 col_params_.instance.instance_key = kInstanceKey; in Init()
382 .Attr("instance_key", params.instance.instance_key) in GetCollectiveBcastSend()
400 .Attr("instance_key", params.instance.instance_key) in GetCollectiveBcastRecv()
564 col_params_.instance.instance_key = in DeviceInstance()
565 parent_->col_params_.instance.instance_key; in DeviceInstance()
682 strings::StrCat(col_params_.instance.instance_key, ":0:0"); in DoBroadcast()
Dring_gatherer_test.cc181 col_params_.instance.instance_key = kInstanceKey; in Init()
354 .Attr("instance_key", params.instance.instance_key) in GetCollectiveGather()
481 strings::StrCat(col_params_.instance.instance_key, ":0:0"); in DoGather()
537 cp.instance.instance_key = 3; in SetUpCollectiveParams()
Dring_reducer_test.cc203 col_params_.instance.instance_key = kInstanceKey; in Init()
377 .Attr("instance_key", params.instance.instance_key) in GetCollectiveReduce()
511 strings::StrCat(col_params_.instance.instance_key, ":0:0"); in DoReduce()
564 cp.instance.instance_key = 3; in SetUpCollectiveParams()
Ddirect_session_test.cc2279 int instance_key) { in CollectiveFunction() argument
2296 {"instance_key", instance_key}, in CollectiveFunction()
/external/tensorflow/tensorflow/core/framework/
Dcollective.cc60 instance_key = other.instance_key; in operator =()
88 string v = strings::StrCat("CollInstanceParams { instance_key=", instance_key, in ToString()
Dcollective.h81 int32 instance_key = -1; member
/external/tensorflow/tensorflow/python/distribute/
Dcross_device_utils.py280 _thread_local.instance_key = self._instance_key_start
323 v = self._get_thread_local_object().instance_key
324 self._get_thread_local_object().instance_key += 1
357 instance_key = collective_keys.get_instance_key()
367 input_tensors[d], group_size, group_key, instance_key, reduction_op,
/external/tensorflow/tensorflow/core/protobuf/
Dworker.proto516 // by instance_key. Service will respond when all group_size ops have
526 int32 instance_key = 7; field
536 int32 instance_key = 1; field
/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 …end(scope *Scope, input tf.Output, group_size int64, group_key int64, instance_key int64, shape tf…
4080 …e{}{"group_size": group_size, "group_key": group_key, "instance_key": instance_key, "shape": shape}
4093 …her(scope *Scope, input tf.Output, group_size int64, group_key int64, instance_key int64, shape tf…
4097 …e{}{"group_size": group_size, "group_key": group_key, "instance_key": instance_key, "shape": shape}
4121 …uce(scope *Scope, input tf.Output, group_size int64, group_key int64, instance_key int64, merge_op…
4125 …e{}{"group_size": group_size, "group_key": group_key, "instance_key": instance_key, "merge_op": me…
27191 …tRecv(scope *Scope, T tf.DataType, group_size int64, group_key int64, instance_key int64, shape tf…
27195 … T, "group_size": group_size, "group_key": group_key, "instance_key": instance_key, "shape": shape}

12