Searched refs:CreateSendDone (Results 1 – 12 of 12) sorted by relevance
125 builder.AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_F()324 builder.AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_F()352 builder.AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_F()
73 builder_.AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_F()
195 auto send_done = builder.AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_F()
145 true_computation->AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_F()
81 builder.AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_F()
194 while_body->AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_F()
367 auto send_done = builder.AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_F()
756 static std::unique_ptr<HloInstruction> CreateSendDone(
215 instruction = CreateSendDone(operands(0), proto.is_host_transfer()); in CreateFromProto()1129 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateSendDone( in CreateSendDone() function in xla::HloInstruction
1266 auto send_done = builder.AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_P()
1524 HloInstruction::CreateSendDone(operands[0], *is_host_transfer)); in ParseInstructionRhs()
2651 [`HloInstruction::CreateSend` and `HloInstruction::CreateSendDone`](https://www.tensorflow.org/code…