| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_param_resolver_local.h | 47 void CompleteParamsAsync(const string& device, CollectiveParams* cp,
85 void CompleteGroupLocal(const string& device, CollectiveParams* cp,
128 CollectiveParams shared GUARDED_BY(out_mu);
163 void FindInstanceRec(const GroupRec* gr, CollectiveParams* cp,
172 void InitInstanceSharedParams(const GroupRec* gr, const CollectiveParams* cp,
177 const CollectiveParams* cp, InstanceRec* ir,
183 void CompleteDefaultRanking(const GroupRec* gr, const CollectiveParams* cp,
191 CollectiveParams* cp, bool is_source,
199 CollectiveParams* cp,
206 void WaitForGroup(InstanceRec* ir, CollectiveParams* cp, bool is_source,
[all …]
|
| D | hierarchical_tree_broadcaster.h | 38 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
45 Status InitializeInstanceBeforeGroupDiscovery(CollectiveParams*) override { in InitializeInstanceBeforeGroupDiscovery() argument
57 static int TreeRecvFrom(const CollectiveParams& cp, int subdiv);
61 static void TreeSendTo(const CollectiveParams& cp, int subdiv,
83 const CollectiveParams* col_params_; // Not owned
|
| D | base_collective_executor.h | 110 void ExecuteAsync(OpKernelContext* ctx, const CollectiveParams& col_params,
113 void CompleteParamsAsync(const string& device, CollectiveParams* cp,
148 void WaitForDependencies(const CollectiveParams& col_params) override;
152 void Launched(const CollectiveParams& col_params) override;
166 Status CreateCollective(const CollectiveParams& col_params,
169 bool CheckDependencies(const CollectiveParams& col_params)
|
| D | collective_param_resolver_local_test.cc | 49 const CollectiveParams& shared_cp, in RunCompleteDefaultRanking()
53 CollectiveParams cp; in RunCompleteDefaultRanking()
84 CollectiveParams cp; in TEST_F()
158 CollectiveParams cps[NUM_DEVS]; in TEST_F()
162 CollectiveParams* cp = &cps[i]; in TEST_F()
205 CollectiveParams* cp) { in InitializeCollectiveParamsForBroadcast()
222 CollectiveParams cps[NUM_DEVS]; in TEST_F()
226 CollectiveParams* cp = &cps[i]; in TEST_F()
260 CollectiveParams cps[NUM_DEVS]; in TEST_F()
264 CollectiveParams* cp = &cps[i]; in TEST_F()
|
| D | collective_param_resolver_local.cc | 58 const string& device, CollectiveParams* cp, const GroupRecCallback& done) { in CompleteGroupLocal()
293 CollectiveParams* cp, const std::vector<DeviceLocality>& localities) { in EstablishGlobalRank()
321 void SetDevPerTask(CollectiveParams* cp) { in SetDevPerTask()
352 void SortDevicesAndTasks(CollectiveParams* cp) { in SortDevicesAndTasks()
382 CollectiveParams* cp) { in CompleteTaskIsLocal()
390 CollectiveParams* cp) { in SetDefaultRank()
401 const GroupRec* gr, const CollectiveParams* cp, InstanceRec* ir, in InitInstanceSharedParams()
460 const GroupRec* gr, const CollectiveParams* cp, InstanceRec* ir, in CompleteDefaultRanking()
502 const GroupRec* gr, CollectiveParams* cp, const InstanceRecCallback& done) { in FindInstanceRec()
536 const GroupRec* gr, const CollectiveParams* cp, InstanceRec* ir, in CallInitInstanceSharedParams()
[all …]
|
| D | base_collective_executor.cc | 219 const CollectiveParams& col_params, in ExecuteAsync()
277 const string& device, CollectiveParams* cp, CancellationManager* cancel_mgr, in CompleteParamsAsync()
284 const CollectiveParams& col_params, in CreateCollective()
312 const CollectiveParams& col_params) { in CheckDependencies()
325 const CollectiveParams& col_params) { in WaitForDependencies()
333 void BaseCollectiveExecutor::Launched(const CollectiveParams& col_params) { in Launched()
|
| D | ring_alg.h | 38 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
45 Status InitializeInstanceBeforeGroupDiscovery(CollectiveParams*) override { in InitializeInstanceBeforeGroupDiscovery() argument
111 const CollectiveParams* col_params_; // Not owned
|
| D | hierarchical_tree_broadcaster_test.cc | 59 CollectiveParams cp; \
372 const CollectiveParams& params, Tensor* input, in GetCollectiveBcastSend()
390 const CollectiveParams& params, const TensorShape& shape, in GetCollectiveBcastRecv()
428 const CollectiveParams& cp = instances_[0]->col_params_; in RunTest()
516 CollectiveParams* cp, in RunSubdivPermsTest()
534 void PrepColParamsForSubdivPermsTest(CollectiveParams* cp, int num_tasks, in PrepColParamsForSubdivPermsTest()
704 CollectiveParams col_params_;
717 CollectiveParams col_params_;
728 CollectiveParams cp; in TEST_F()
748 CollectiveParams cp; in TEST_F()
[all …]
|
| D | device_resolver_local_test.cc | 49 CollectiveParams cp; in TEST_F()
68 CollectiveParams cp; in TEST_F()
|
| D | ring_reducer_test.cc | 362 std::unique_ptr<OpKernel> GetCollectiveReduce(const CollectiveParams& params, in GetCollectiveReduce()
385 CollectiveParams* cp, in RunSubdivPermsTest()
535 CollectiveParams col_params_;
548 CollectiveParams col_params_;
556 CollectiveParams SetUpCollectiveParams(const int num_devs_per_task, in SetUpCollectiveParams()
558 CollectiveParams cp; in SetUpCollectiveParams()
585 CollectiveParams cp = SetUpCollectiveParams(kNumDevsPerTask, kNumTasks); in TEST_F()
618 CollectiveParams cp = SetUpCollectiveParams(kNumDevsPerTask, kNumTasks); in TEST_F()
649 CollectiveParams cp = SetUpCollectiveParams(kNumDevsPerTask, kNumTasks); in TEST_F()
|
| D | ring_gatherer_test.cc | 342 std::unique_ptr<OpKernel> GetCollectiveGather(const CollectiveParams& params, in GetCollectiveGather()
362 CollectiveParams* cp, in RunSubdivPermsTest()
508 CollectiveParams col_params_;
521 CollectiveParams col_params_;
529 CollectiveParams SetUpCollectiveParams(const int num_devs_per_task, in SetUpCollectiveParams()
531 CollectiveParams cp; in SetUpCollectiveParams()
558 CollectiveParams cp = SetUpCollectiveParams(kNumDevsPerTask, kNumTasks); in TEST_F()
|
| D | ring_gatherer.h | 36 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
|
| D | ring_reducer.h | 42 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
|
| D | hierarchical_tree_broadcaster.cc | 77 CollectiveParams* col_params) { in InitializeCollectiveParams()
217 int HierarchicalTreeBroadcaster::TreeRecvFrom(const CollectiveParams& cp, in TreeRecvFrom()
236 void HierarchicalTreeBroadcaster::TreeSendTo(const CollectiveParams& cp, in TreeSendTo()
|
| D | collective_util.h | 33 string SubdivPermDebugString(const CollectiveParams& col_params);
|
| D | collective_util.cc | 54 string SubdivPermDebugString(const CollectiveParams& col_params) { in SubdivPermDebugString()
|
| D | ring_alg.cc | 108 Status GenerateSubdivsInCollectiveParams(CollectiveParams* col_params) { in GenerateSubdivsInCollectiveParams()
164 Status RingAlg::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 111 struct CollectiveParams { struct
159 virtual void CompleteParamsAsync(const string& device, CollectiveParams* cp,
262 const CollectiveParams& col_params, in ExecuteAsync()
269 virtual void CompleteParamsAsync(const string& device, CollectiveParams* cp, in CompleteParamsAsync()
288 virtual void WaitForDependencies(const CollectiveParams& col_params) {} in WaitForDependencies()
292 virtual void Launched(const CollectiveParams& col_params) {} in Launched()
343 const CollectiveParams& col_params, const string& exec_key,
352 const CollectiveParams& col_params;
378 virtual Status InitializeCollectiveParams(CollectiveParams* col_params) = 0;
392 CollectiveParams* col_params) = 0;
|
| D | collective.cc | 137 string CollectiveParams::ToString() const { in ToString()
160 const CollectiveParams& col_params, in CollectiveContext()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed.h | 34 void CompleteParamsAsync(const string& device, CollectiveParams* cp,
62 void CompleteGroupDistributed(const string& device, CollectiveParams* cp,
71 void UpdateInstanceCache(const GroupRec* gr, CollectiveParams* cp,
80 CollectiveParams* cp,
|
| D | collective_param_resolver_distributed.cc | 96 const string& device, CollectiveParams* cp, CancellationManager* cancel_mgr, in CompleteParamsAsync()
115 CollectiveParams cp; in CompleteGroupAsync()
147 CollectiveParams* cp = new CollectiveParams; in CompleteInstanceAsync()
254 const string& device, CollectiveParams* cp, CancellationManager* cancel_mgr, in CompleteGroupDistributed()
291 const GroupRec* gr, CollectiveParams* cp, in UpdateInstanceCache()
363 const string& device, const GroupRec* gr, CollectiveParams* cp, in CompleteInstanceDistributed()
|
| D | collective_param_resolver_distributed_test.cc | 190 cp_.push_back(CollectiveParams()); in DefineCollectiveParams()
191 CollectiveParams& cp = cp_.back(); in DefineCollectiveParams()
232 CollectiveParams* cp = &cp_[idx]; in IssueRequest()
290 std::vector<CollectiveParams> cp_;
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer.h | 29 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
36 CollectiveParams* col_params) override;
43 const CollectiveParams* col_params_; // Not owned
|
| D | collective_nccl_reducer.cc | 31 Status NcclReducer::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()
53 CollectiveParams* col_params) { in InitializeInstanceBeforeGroupDiscovery()
|
| D | collective_nccl_reducer_test.cc | 194 std::unique_ptr<OpKernel> GetCollectiveReduce(const CollectiveParams& params, in GetCollectiveReduce()
310 CollectiveParams col_params_;
319 CollectiveParams col_params_;
|