Searched defs:recv_done (Results 1 – 15 of 15) sorted by relevance
108 HloInstruction* recv_done = in GlobalPredecessors() local180 const HloInstruction* recv_done = instruction->users().front(); in GlobalSuccessors() local188 HloInstruction* recv_done = in GlobalSuccessors() local
199 auto recv_done = builder.AddInstruction(HloInstruction::CreateRecvDone(recv)); in TEST_F() local
142 Status LogicalBufferAnalysis::HandleRecvDone(HloInstruction* recv_done) { in HandleRecvDone()
110 HloInstruction* recv_done = nullptr; member
197 Status HandleRecvDone(HloInstructionPtr recv_done) override { in HandleRecvDone()
713 Status ShapeVerifier::HandleRecvDone(HloInstruction* recv_done) { in HandleRecvDone()1156 const HloInstruction* recv_done = instruction->users().front(); in VerifySendsAndRecvs() local
334 auto recv_done = builder.AddInstruction(HloInstruction::CreateRecvDone(recv)); in TEST_F() local
287 Status TuplePointsToAnalysis::HandleRecvDone(HloInstruction* recv_done) { in HandleRecvDone()
340 bool HloDataflowAnalysis::UpdateRecvDoneValueSet(HloInstruction* recv_done) { in UpdateRecvDoneValueSet()
364 auto recv_done = builder.AddInstruction(HloInstruction::CreateRecvDone(recv)); in TEST_F() local
1207 auto recv_done = builder.AddInstruction(HloInstruction::CreateRecvDone(recv)); in TEST_P() local
101 void Start(std::function<void()> recv_done) override { in Start()137 void StartRTCall(std::function<void()> recv_done) { in StartRTCall()143 const Status& s) { in StartRTCall()
56 void Start(std::function<void()> recv_done) override { in Start()
184 auto recv_done = [this, &state, i](const Status& status, in TEST_F() local
2085 Status IrEmitter::HandleRecvDone(HloInstruction* recv_done) { in HandleRecvDone()