Searched refs:RingField (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_alg.h | 67 struct RingField { struct
85 virtual void InitRingField(RingField* rf, int chunk_idx, int subdiv_idx, argument
87 void AdvanceToSecondPass(RingField* rf);
88 void DispatchSend(RingField* rf, const StatusCallback& done);
89 void DispatchRecv(RingField* rf, const StatusCallback& done);
98 void Enqueue(RingField* rf);
99 RingField* Dequeue();
105 std::deque<RingField*> deque_ GUARDED_BY(pcq_mu_);
120 std::vector<RingField> rfv_;
|
| D | ring_alg.cc | 76 void RingAlg::PCQueue::Enqueue(RingField* rf) { in Enqueue()
84 RingAlg::RingField* RingAlg::PCQueue::Dequeue() { in Dequeue()
93 RingField* rf = deque_.front(); in Dequeue()
309 void RingAlg::InitRingField(RingField* rf, int chunk_idx, int subdiv_idx, in InitRingField()
352 void RingAlg::AdvanceToSecondPass(RingField* rf) { in AdvanceToSecondPass()
369 string RingAlg::RingField::DebugString() const { in DebugString()
380 void RingAlg::DispatchSend(RingField* rf, const StatusCallback& done) { in DispatchSend()
398 void RingAlg::DispatchRecv(RingField* rf, const StatusCallback& done) { in DispatchRecv()
|
| D | ring_reducer.cc | 145 void RingReducer::InitRingField(RingField* rf, int chunk_idx, int subdiv_idx, in InitRingField()
200 RingField* rf = ready_queue.Dequeue(); in RunAsyncParts()
304 RingField* rf = ready_queue.Dequeue(); in RunAsyncParts()
|
| D | ring_gatherer.cc | 164 RingField* rf = ready_queue.Dequeue(); in RunAsyncParts()
242 RingField* rf = ready_queue.Dequeue(); in RunAsyncParts()
|
| D | ring_reducer.h | 45 void InitRingField(RingField* rf, int chunk_idx, int subdiv_idx,
|