Searched refs:send_recv_instr (Results 1 – 2 of 2) sorted by relevance
205 const HloSendRecvInstruction* send_recv_instr = in IsChannelInstruction() local207 CHECK(send_recv_instr != nullptr); in IsChannelInstruction()208 return !send_recv_instr->is_host_transfer(); in IsChannelInstruction()
394 const HloSendRecvInstruction* send_recv_instr = in IsHostSendRecv() local396 return send_recv_instr != nullptr && send_recv_instr->is_host_transfer(); in IsHostSendRecv()404 const HloSendRecvInstruction* send_recv_instr = in BuildHostChannelConstraints() local406 if (send_recv_instr == nullptr || !send_recv_instr->is_host_transfer()) { in BuildHostChannelConstraints()414 ShapeUtil::GetTupleElementShape(send_recv_instr->shape(), 0); in BuildHostChannelConstraints()418 *send_recv_instr->channel_id(), data_shape.layout()); in BuildHostChannelConstraints()422 << send_recv_instr->ToString(); in BuildHostChannelConstraints()