Searched refs:send_recv_instr (Results 1 – 2 of 2) sorted by relevance
188 const HloSendRecvInstruction* send_recv_instr = in IsChannelInstruction() local190 CHECK(send_recv_instr != nullptr); in IsChannelInstruction()191 return !send_recv_instr->is_host_transfer(); in IsChannelInstruction()
384 const HloSendRecvInstruction* send_recv_instr = in IsHostSendRecv() local386 return send_recv_instr != nullptr && send_recv_instr->is_host_transfer(); in IsHostSendRecv()394 const HloSendRecvInstruction* send_recv_instr = in BuildHostChannelConstraints() local396 if (send_recv_instr == nullptr || !send_recv_instr->is_host_transfer()) { in BuildHostChannelConstraints()404 ShapeUtil::GetTupleElementShape(send_recv_instr->shape(), 0); in BuildHostChannelConstraints()408 send_recv_instr->channel_id(), data_shape.layout()); in BuildHostChannelConstraints()412 << send_recv_instr->ToString(); in BuildHostChannelConstraints()