Searched defs:send_done (Results 1 – 11 of 11) sorted by relevance
195 auto send_done = builder.AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_F() local
109 HloInstruction* send_done = nullptr; member
203 Status HandleSendDone(HloInstructionPtr send_done) override { in HandleSendDone()
702 Status ShapeVerifier::HandleSendDone(HloInstruction* send_done) { in HandleSendDone()1147 const HloInstruction* send_done = instruction->users().front(); in VerifySendsAndRecvs() local
337 auto send_done = builder.AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_F() local
339 auto send_done = builder.AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_F() local
1181 auto send_done = builder.AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_P() local
660 bool send_done = false; in WaitForReply() local
542 auto send_done = tx_task_runner.CreateCheckpoint("send_done"); in TEST_F() local
107 volatile int send_wait, reap_wait, send_done, reap_done; member
1909 Status IrEmitter::HandleSendDone(HloInstruction* send_done) { in HandleSendDone()