Home
last modified time | relevance | path

Searched refs:send_recv_instr (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_group_metadata.cc205 const HloSendRecvInstruction* send_recv_instr = in IsChannelInstruction() local
207 CHECK(send_recv_instr != nullptr); in IsChannelInstruction()
208 return !send_recv_instr->is_host_transfer(); in IsChannelInstruction()
Dlayout_assignment.cc394 const HloSendRecvInstruction* send_recv_instr = in IsHostSendRecv() local
396 return send_recv_instr != nullptr && send_recv_instr->is_host_transfer(); in IsHostSendRecv()
404 const HloSendRecvInstruction* send_recv_instr = in BuildHostChannelConstraints() local
406 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()