| 663 | } |
| 664 | |
| 665 | XlaOp XlaBuilder::AsyncOutSend(const XlaOp& operand, |
| 666 | const Shape& shape_with_layout, |
| 667 | const string& rendezvous_key) { |
| 668 | return ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { |
| 669 | HloInstructionProto instr; |
| 670 | *instr.mutable_shape() = ShapeUtil::MakeNil().ToProto(); |
| 671 | *instr.mutable_async_out_send_shape() = shape_with_layout.ToProto(); |
| 672 | instr.set_rendezvous_key(rendezvous_key); |
| 673 | return AddInstruction(std::move(instr), HloOpcode::kAsyncOutSend, |
| 674 | {operand}); |
| 675 | }); |
| 676 | } |
| 677 | |
| 678 | XlaOp XlaBuilder::Iota(const Shape& shape, int64 iota_dimension) { |
| 679 | return ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { |
no test coverage detected