Searched defs:send_done (Results 1 – 11 of 11) sorted by relevance
107 HloInstruction* send_done = nullptr; member
196 auto send_done = builder.AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_F() local
253 Status HandleSendDone(HloInstructionPtr send_done) override { in HandleSendDone()
1534 Status ShapeVerifier::HandleSendDone(HloInstruction* send_done) { in HandleSendDone()2124 const HloInstruction* send_done = instruction->users().front(); in VerifyChannels() local
1193 auto send_done = builder.AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_P() local
315 HloInstruction* send_done, in RemoveSendDone()
569 bool send_done = false; in WaitForReply() local
660 bool send_done = false; in WaitForReply() local
107 volatile int send_wait, reap_wait, send_done, reap_done; member
372 auto send_done = tx_task_runner.CreateCheckpoint("send_done"); in TEST_F() local
1934 Status IrEmitter::HandleSendDone(HloInstruction* send_done) { in HandleSendDone()