Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.h51 void CompleteParamsAsync(const DeviceAttributes& device, CollectiveParams* cp,
111 CollectiveParams* shared;
127 : shared(new CollectiveParams()), source_rank(-1), known_count(0) {} in InstanceRec()
139 InstanceRec* GetOrCreateInstanceRec(CollectiveParams* cp, bool* created)
147 void InitInstanceSharedParams(const CollectiveParams* cp, InstanceRec* ir);
155 void CompleteInstanceLocal(const string& device, CollectiveParams* cp,
162 CollectiveParams* cp,
169 void WaitForGroup(InstanceRec* ir, CollectiveParams* cp, const IRConsumer& f)
174 Status GetLocalDeviceLocalities(const CollectiveParams& cp,
179 void SetDefaultRank(const string& device, CollectiveParams* cp);
[all …]
Dbase_collective_executor.h112 void ExecuteAsync(OpKernelContext* ctx, const CollectiveParams* col_params,
115 void CompleteParamsAsync(const DeviceAttributes& device, CollectiveParams* cp,
131 void WaitForDependencies(const CollectiveParams& col_params) override;
135 void UnblockDependencies(const CollectiveParams& col_params) override;
153 Status CreateCollective(const CollectiveParams& col_params,
156 bool CheckDependencies(const CollectiveParams& col_params)
Dhierarchical_tree_broadcaster.h38 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
53 static int TreeRecvFrom(const CollectiveParams& cp, int subdiv);
57 static void TreeSendTo(const CollectiveParams& cp, int subdiv,
79 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()
63 const CollectiveParams* col_params_; // Not owned
Dall_to_all.h36 Status InitializeCollectiveParams(CollectiveParams* col_params) override { in InitializeCollectiveParams()
47 const CollectiveParams* col_params_; // Not owned
Dbase_collective_executor.cc273 const CollectiveParams* col_params, in ExecuteAsync()
357 const DeviceAttributes& device, CollectiveParams* cp, in CompleteParamsAsync()
406 const CollectiveParams& col_params, in CreateCollective()
457 const CollectiveParams& col_params) { in CheckDependencies()
470 const CollectiveParams& col_params) { in WaitForDependencies()
479 const CollectiveParams& col_params) { in UnblockDependencies()
Dring_alg.h38 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
107 const CollectiveParams* col_params_; // Not owned
Dcollective_test_util.h92 core::RefCountPtr<CollectiveParams> CreateCollectiveParams(
101 Status RunCollective(CollectiveTestEnv* test_env, CollectiveParams* col_params,
Dcollective_param_resolver_local.cc64 const char* GetCollectiveName(const CollectiveParams* cp, bool nccl) { in GetCollectiveName()
529 CollectiveParams* cp) { in SetDefaultRank()
543 const CollectiveParams* cp, InstanceRec* ir) { in InitInstanceSharedParams()
598 CollectiveParamResolverLocal::GetOrCreateInstanceRec(CollectiveParams* cp, in GetOrCreateInstanceRec()
660 const DeviceAttributes& device, CollectiveParams* cp, in CompleteParamsAsync()
697 void CollectiveParamResolverLocal::AssignCollectiveType(CollectiveParams* cp) { in AssignCollectiveType()
716 const string& device, CollectiveParams* cp, const StatusCallback& done) { in CompleteInstanceLocal()
740 const string& device, CollectiveParams* cp, InstanceRec* ir, in CompleteInstanceFromInitializedIRec()
803 CollectiveParams* cp, in WaitForGroup()
Dhierarchical_tree_broadcaster_test.cc62 auto* cp = new CollectiveParams(); \
247 core::RefCountPtr<CollectiveParams> col_params_;
259 CollectiveParams* cp, in RunSubdivPermsTest()
350 auto* cp = new CollectiveParams(); in TEST_F()
Dcollective_test_util.cc184 core::RefCountPtr<CollectiveParams> CreateCollectiveParams( in CreateCollectiveParams()
190 core::RefCountPtr<CollectiveParams> col_params(new CollectiveParams()); in CreateCollectiveParams()
291 Status RunCollective(CollectiveTestEnv* test_env, CollectiveParams* col_params, in RunCollective()
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);
Dhierarchical_tree_broadcaster.cc79 CollectiveParams* col_params) { in InitializeCollectiveParams()
218 int HierarchicalTreeBroadcaster::TreeRecvFrom(const CollectiveParams& cp, in TreeRecvFrom()
237 void HierarchicalTreeBroadcaster::TreeSendTo(const CollectiveParams& cp, in TreeSendTo()
/external/tensorflow/tensorflow/core/framework/
Dcollective.h144 struct CollectiveParams : public core::RefCounted { struct
190 CollectiveParams* cp,
316 const CollectiveParams* col_params, in ExecuteAsync()
324 CollectiveParams* cp, in CompleteParamsAsync()
358 virtual void WaitForDependencies(const CollectiveParams& col_params) {} in WaitForDependencies()
362 virtual void UnblockDependencies(const CollectiveParams& col_params) {} in UnblockDependencies()
397 core::IntrusivePtr<const CollectiveParams> col_params;
410 const CollectiveParams* col_params, const string& exec_key,
442 virtual Status InitializeCollectiveParams(CollectiveParams* col_params) = 0;
Dcollective.cc137 string CollectiveParams::ToString() const { in ToString()
158 OpKernelContext::Params* op_params, const CollectiveParams* col_params, in CollectiveContext()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed.h37 void CompleteParamsAsync(const DeviceAttributes& device, CollectiveParams* cp,
78 Status UpdateInstanceCache(CollectiveParams* cp,
85 void CompleteInstanceDistributed(const string& device, CollectiveParams* cp,
Dcollective_param_resolver_distributed_test.cc180 CollectiveParams* CreateCollectiveParams(int num_workers, int num_devices, in CreateCollectiveParams()
186 auto* cp = new CollectiveParams(); in CreateCollectiveParams()
219 CollectiveParams* cp = cp_[device_name]; in IssueRequest()
274 void ValidateDeviceResolver(const CollectiveParams& cp, const string& task) { in ValidateDeviceResolver()
315 absl::flat_hash_map<string, CollectiveParams*> cp_;
Dcollective_param_resolver_distributed.cc108 const DeviceAttributes& device, CollectiveParams* cp, in CompleteParamsAsync()
158 CollectiveParams* cp = new CollectiveParams; in CompleteInstanceAsync()
328 CollectiveParams* cp, const CompleteInstanceResponse& resp) { in UpdateInstanceCache()
364 const string& device, CollectiveParams* cp, CancellationManager* cancel_mgr, in CompleteInstanceDistributed()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl.h29 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
39 const CollectiveParams* col_params_; // Not owned
Dcollective_ops.cc69 : AsyncOpKernel(c), name_(name()), col_params_(new CollectiveParams()) {} in CollectiveOpV1Kernel()
146 CollectiveParams* col_params_;
588 Status FillCollectiveParams(CollectiveParams* col_params, in FillCollectiveParams()
626 void Run(OpKernelContext* c, CollectiveParams* col_params, in Run()
716 auto col_params = new CollectiveParams(); in ComputeAsync()
767 auto col_params = new CollectiveParams(); in ComputeAsync()
813 auto col_params = new CollectiveParams(); in ComputeAsync()
862 auto col_params = new CollectiveParams(); in ComputeAsync()
1065 Status FillCollectiveParams(CollectiveParams* col_params, in FillCollectiveParams()
1100 void Run(OpKernelContext* c, CollectiveParams* col_params, in Run()
[all …]
Dcollective_nccl_test.cc132 col_params_ = new CollectiveParams(); in Init()
231 const CollectiveParams& params, Tensor* input, DeviceBase* device) { in GetCollectiveReduceOpKernel()
255 col_params_(new CollectiveParams()) { in DeviceInstance()
428 CollectiveParams* col_params_;
444 CollectiveParams* col_params_;
Dcollective_nccl.cc29 Status NcclBase::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()

12