Searched defs:send_done (Results 1 – 8 of 8) sorted by relevance
306 const HloInstruction* send_done = send->users().front(); in HandleSend() local314 Status ShapeVerifier::HandleSendDone(HloInstruction* send_done) { in HandleSendDone()
188 Status HandleSendDone(HloInstructionPtr send_done) override { in HandleSendDone()
323 auto send_done = builder.AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_F() local
335 auto send_done = builder.AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_F() local
1163 auto send_done = builder.AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_P() local
513 auto send_done = tx_task_runner.CreateCheckpoint("send_done"); in TEST_F() local
109 volatile int send_wait, reap_wait, send_done, reap_done; member
1820 Status IrEmitter::HandleSendDone(HloInstruction* send_done) { in HandleSendDone()