Searched refs:InstanceRec (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_param_resolver_local.h | 96 struct InstanceRec;
98 typedef std::function<void(InstanceRec*)> IRConsumer;
99 struct InstanceRec { struct
117 InstanceRec() in InstanceRec() function
119 ~InstanceRec() { shared->Unref(); } in ~InstanceRec() argument
130 InstanceRec* GetOrCreateInstanceRec(const GroupRec* gr, CollectiveParams* cp,
140 InstanceRec* ir) TF_LOCKS_EXCLUDED(gr->mu);
159 InstanceRec* ir, bool is_source,
165 void WaitForGroup(InstanceRec* ir, CollectiveParams* cp, bool is_source,
196 gtl::FlatMap<int32, gtl::FlatMap<int32, std::unique_ptr<InstanceRec>>>
|
| D | collective_param_resolver_local.cc | 555 const GroupRec* gr, const CollectiveParams* cp, InstanceRec* ir) { in InitInstanceSharedParams()
599 CollectiveParamResolverLocal::InstanceRec*
604 InstanceRec* irec = nullptr; in GetOrCreateInstanceRec()
616 irec = new InstanceRec; in GetOrCreateInstanceRec()
702 InstanceRec* ir = GetOrCreateInstanceRec(gr, cp, &created_irec); in CompleteInstanceLocal()
721 InstanceRec* ir, bool is_source, const StatusCallback& done) { in CompleteInstanceFromInitializedIRec()
764 [col_impl, ir, device, cp, done](InstanceRec* irec) { in CompleteInstanceFromInitializedIRec()
784 void CollectiveParamResolverLocal::WaitForGroup(InstanceRec* ir, in WaitForGroup()
866 std::vector<InstanceRec*> instances; in StartAbortLocal()
875 for (InstanceRec* ir : instances) { in StartAbortLocal()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed.cc | 208 InstanceRec* ir = GetOrCreateInstanceRec(gr, cp, &created_irec); in CompleteInstanceAsync()
346 InstanceRec* ir = GetOrCreateInstanceRec(gr, cp, &created_irec); in UpdateInstanceCache()
|