Searched refs:recv_instr (Results 1 – 1 of 1) sorted by relevance
2387 HloInstructionProto recv_instr; in RecvWithToken() local2388 *recv_instr.mutable_shape() = in RecvWithToken()2392 recv_instr.set_channel_id(handle.handle()); in RecvWithToken()2393 TF_ASSIGN_OR_RETURN(XlaOp recv, AddInstruction(std::move(recv_instr), in RecvWithToken()2473 HloInstructionProto recv_instr; in RecvFromHost() local2474 *recv_instr.mutable_shape() = in RecvFromHost()2478 recv_instr.set_channel_id(handle.handle()); in RecvFromHost()2479 recv_instr.set_is_host_transfer(true); in RecvFromHost()2480 TF_ASSIGN_OR_RETURN(XlaOp recv, AddInstruction(std::move(recv_instr), in RecvFromHost()