Home
last modified time | relevance | path

Searched refs:CollectiveParams (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.h50 void CompleteParamsAsync(const DeviceAttributes& device, CollectiveParams* cp,
87 void CompleteGroupLocal(const DeviceAttributes& device, CollectiveParams* cp,
102 CollectiveParams* shared;
118 : shared(new CollectiveParams()), source_rank(-1), known_count(0) {} in InstanceRec()
130 InstanceRec* GetOrCreateInstanceRec(const GroupRec* gr, CollectiveParams* cp,
139 void InitInstanceSharedParams(const GroupRec* gr, const CollectiveParams* cp,
150 CollectiveParams* cp, bool is_source,
158 CollectiveParams* cp,
165 void WaitForGroup(InstanceRec* ir, CollectiveParams* cp, bool is_source,
170 Status GetLocalDeviceLocalities(const CollectiveParams& cp,
[all …]
Dbase_collective_executor.h113 void ExecuteAsync(OpKernelContext* ctx, const CollectiveParams* col_params,
116 void CompleteParamsAsync(const DeviceAttributes& device, CollectiveParams* cp,
132 void WaitForDependencies(const CollectiveParams& col_params) override;
136 void UnblockDependencies(const CollectiveParams& col_params) override;
155 Status CreateCollective(const CollectiveParams& col_params,
158 bool CheckDependencies(const CollectiveParams& col_params)
Dhierarchical_tree_broadcaster.h38 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
59 static int TreeRecvFrom(const CollectiveParams& cp, int subdiv);
63 static void TreeSendTo(const CollectiveParams& cp, int subdiv,
85 const CollectiveParams* col_params_; // Not owned
Dcollective_param_resolver_local_test.cc164 CollectiveParams* cps[NUM_DEVS]; in TEST_F()
168 cps[i] = new CollectiveParams(); in TEST_F()
169 CollectiveParams* cp = cps[i]; in TEST_F()
213 CollectiveParams* cp) { in InitializeCollectiveParamsForBroadcast()
228 CollectiveParams* cps[NUM_DEVS]; in TEST_F()
232 cps[i] = new CollectiveParams(); in TEST_F()
233 CollectiveParams* cp = cps[i]; in TEST_F()
270 CollectiveParams* cps[NUM_DEVS]; in TEST_F()
274 cps[i] = new CollectiveParams(); in TEST_F()
275 CollectiveParams* cp = cps[i]; in TEST_F()
[all …]
Dpermuter.h52 Status InitializeCollectiveParams(CollectiveParams* col_params) override { in InitializeCollectiveParams()
68 const CollectiveParams* col_params_; // Not owned
Dhierarchical_tree_broadcaster_test.cc61 auto* cp = new CollectiveParams(); \
268 col_params_ = new CollectiveParams(); in Init()
386 const CollectiveParams& params, Tensor* input, in GetCollectiveBcastSend()
404 const CollectiveParams& params, const TensorShape& shape, in GetCollectiveBcastRecv()
442 const CollectiveParams& cp = *instances_[0]->col_params_; in RunTest()
521 CollectiveParams* cp, in RunSubdivPermsTest()
540 void PrepColParamsForSubdivPermsTest(CollectiveParams* cp, int num_tasks, in PrepColParamsForSubdivPermsTest()
566 col_params_(new CollectiveParams()) { in DeviceInstance()
705 CollectiveParams* col_params_;
719 CollectiveParams* col_params_;
[all …]
Dring_reducer_test.cc208 col_params_ = new CollectiveParams(); in Init()
373 std::unique_ptr<OpKernel> GetCollectiveReduce(const CollectiveParams& params, in GetCollectiveReduce()
396 CollectiveParams* cp, in RunSubdivPermsTest()
420 col_params_(new CollectiveParams()) { in DeviceInstance()
546 CollectiveParams* col_params_;
562 CollectiveParams* col_params_;
571 CollectiveParams* SetUpCollectiveParams(const int num_devs_per_task, in SetUpCollectiveParams()
573 auto cp = new CollectiveParams(); in SetUpCollectiveParams()
600 CollectiveParams* cp = SetUpCollectiveParams(kNumDevsPerTask, kNumTasks); in TEST_F()
634 CollectiveParams* cp = SetUpCollectiveParams(kNumDevsPerTask, kNumTasks); in TEST_F()
[all …]
Dbase_collective_executor.cc267 const CollectiveParams* col_params, in ExecuteAsync()
347 const DeviceAttributes& device, CollectiveParams* cp, in CompleteParamsAsync()
393 const CollectiveParams& col_params, in CreateCollective()
434 const CollectiveParams& col_params) { in CheckDependencies()
447 const CollectiveParams& col_params) { in WaitForDependencies()
456 const CollectiveParams& col_params) { in UnblockDependencies()
Dcollective_param_resolver_local.cc59 const char* GetCollectiveName(const CollectiveParams* cp, bool nccl) { in GetCollectiveName()
88 const DeviceAttributes& device, CollectiveParams* cp, in CompleteGroupLocal()
536 CollectiveParams* cp) { in CompleteTaskIsLocal()
544 CollectiveParams* cp) { in SetDefaultRank()
555 const GroupRec* gr, const CollectiveParams* cp, InstanceRec* ir) { in InitInstanceSharedParams()
601 CollectiveParams* cp, in GetOrCreateInstanceRec()
640 const DeviceAttributes& device, CollectiveParams* cp, in CompleteParamsAsync()
668 void CollectiveParamResolverLocal::AssignCollectiveType(CollectiveParams* cp) { in AssignCollectiveType()
686 const string& device, const GroupRec* gr, CollectiveParams* cp, in CompleteInstanceLocal()
720 const string& device, const GroupRec* gr, CollectiveParams* cp, in CompleteInstanceFromInitializedIRec()
[all …]
Dring_gatherer_test.cc185 col_params_ = new CollectiveParams(); in Init()
351 std::unique_ptr<OpKernel> GetCollectiveGather(const CollectiveParams& params, in GetCollectiveGather()
371 CollectiveParams* cp, in RunSubdivPermsTest()
394 col_params_(new CollectiveParams()) { in DeviceInstance()
516 CollectiveParams* col_params_;
530 CollectiveParams* col_params_;
539 CollectiveParams* SetUpCollectiveParams(const int num_devs_per_task, in SetUpCollectiveParams()
541 auto* cp = new CollectiveParams(); in SetUpCollectiveParams()
568 CollectiveParams* cp = SetUpCollectiveParams(kNumDevsPerTask, kNumTasks); in TEST_F()
Dring_alg.h38 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
113 const CollectiveParams* col_params_; // Not owned
Dring_gatherer.h36 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
Dring_reducer.h42 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
Dcollective_util.h33 string SubdivPermDebugString(const CollectiveParams& col_params);
Dpermuter_test.cc175 col_params_ = new CollectiveParams(); in Init()
327 col_params_(new CollectiveParams()) { in DeviceInstance()
423 CollectiveParams* col_params_;
436 CollectiveParams* col_params_;
Dhierarchical_tree_broadcaster.cc78 CollectiveParams* col_params) { in InitializeCollectiveParams()
217 int HierarchicalTreeBroadcaster::TreeRecvFrom(const CollectiveParams& cp, in TreeRecvFrom()
236 void HierarchicalTreeBroadcaster::TreeSendTo(const CollectiveParams& cp, in TreeSendTo()
Dtest_collective_executor_mgr.h39 void CompleteParamsAsync(const DeviceAttributes& device, CollectiveParams* cp, in CompleteParamsAsync()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed.h37 void CompleteParamsAsync(const DeviceAttributes& device, CollectiveParams* cp,
68 CollectiveParams* cp,
78 Status UpdateInstanceCache(const GroupRec* gr, CollectiveParams* cp,
86 CollectiveParams* cp,
Dcollective_param_resolver_distributed.cc108 const DeviceAttributes& device, CollectiveParams* cp, in CompleteParamsAsync()
140 auto* cp = new CollectiveParams(); in CompleteGroupAsync()
186 CollectiveParams* cp = new CollectiveParams; in CompleteInstanceAsync()
287 const DeviceAttributes& device, CollectiveParams* cp, in CompleteGroupDistributed()
342 const GroupRec* gr, CollectiveParams* cp, in UpdateInstanceCache()
379 const string& device, const GroupRec* gr, CollectiveParams* cp, in CompleteInstanceDistributed()
Dcollective_param_resolver_distributed_test.cc191 CollectiveParams* CreateCollectiveParams(int num_workers, int num_devices, in CreateCollectiveParams()
195 auto* cp = new CollectiveParams(); in CreateCollectiveParams()
227 CollectiveParams* cp = cp_[device_name]; in IssueRequest()
281 void ValidateDeviceResolver(const CollectiveParams& cp, const string& task) { in ValidateDeviceResolver()
318 absl::flat_hash_map<string, CollectiveParams*> cp_;
365 Status InitializeCollectiveParams(CollectiveParams*) override { in InitializeCollectiveParams() argument
/external/tensorflow/tensorflow/core/framework/
Dcollective.h135 struct CollectiveParams : public core::RefCounted { struct
181 CollectiveParams* cp,
301 const CollectiveParams* col_params, in ExecuteAsync()
309 CollectiveParams* cp, in CompleteParamsAsync()
331 virtual void WaitForDependencies(const CollectiveParams& col_params) {} in WaitForDependencies()
335 virtual void UnblockDependencies(const CollectiveParams& col_params) {} in UnblockDependencies()
370 const CollectiveParams* col_params; // Not owned
383 const CollectiveParams* col_params, const string& exec_key,
413 virtual Status InitializeCollectiveParams(CollectiveParams* col_params) = 0;
Dcollective.cc150 string CollectiveParams::ToString() const { in ToString()
172 OpKernelContext::Params* op_params, const CollectiveParams* col_params, in CollectiveContext()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl.h29 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
43 const CollectiveParams* col_params_; // Not owned
Dcollective_nccl_test.cc131 col_params_ = new CollectiveParams(); in Init()
229 const CollectiveParams& params, Tensor* input, DeviceBase* device) { in GetCollectiveReduceOpKernel()
253 col_params_(new CollectiveParams()) { in DeviceInstance()
426 CollectiveParams* col_params_;
441 CollectiveParams* col_params_;
Dcollective_nccl.cc29 Status NcclBase::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()

12