Home
last modified time | relevance | path

Searched refs:InstanceRec (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.h90 struct InstanceRec;
92 typedef std::function<void(InstanceRec*)> IRConsumer;
93 struct InstanceRec { struct
143 InstanceRec() in InstanceRec() argument
161 typedef std::function<void(const Status& s, InstanceRec* ir)>
173 InstanceRec* ir, const StatusCallback& done)
177 const CollectiveParams* cp, InstanceRec* ir,
184 InstanceRec* ir,
200 InstanceRec* ir, bool is_source,
206 void WaitForGroup(InstanceRec* ir, CollectiveParams* cp, bool is_source,
[all …]
Dcollective_param_resolver_local.cc36 void CollectiveParamResolverLocal::InstanceRec::WaitForOutMu(mutex_lock& lock) { in WaitForOutMu()
401 const GroupRec* gr, const CollectiveParams* cp, InstanceRec* ir, in InitInstanceSharedParams()
460 const GroupRec* gr, const CollectiveParams* cp, InstanceRec* ir, in CompleteDefaultRanking()
491 const InstanceRecCallback& done, InstanceRec* irec) { in CallbackWithStatus()
503 InstanceRec* irec = nullptr; in FindInstanceRec()
515 irec->init_waiters.push_back([this, done](InstanceRec* irec) { in FindInstanceRec()
523 irec = new InstanceRec; in FindInstanceRec()
536 const GroupRec* gr, const CollectiveParams* cp, InstanceRec* ir, in CallInitInstanceSharedParams()
646 InstanceRec* ir) { in CompleteInstanceLocal()
658 InstanceRec* ir, bool is_source, const StatusCallback& done) { in CompleteInstanceFromInitializedIRec()
[all …]
Dcollective_param_resolver_local_test.cc55 CollectiveParamResolverLocal::InstanceRec ir; in RunCompleteDefaultRanking()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed.cc184 const Status& fi_status, InstanceRec* ir) { in CompleteInstanceAsync()
293 using InstanceRecPointer = InstanceRec*; in UpdateInstanceCache()
306 InstanceRec* ir = *irp; in UpdateInstanceCache()
356 gr, cp, [this, irp, continue_with_ir](const Status s, InstanceRec* irec) { in UpdateInstanceCache()