Home
last modified time | relevance | path

Searched refs:HloSendRecvInstruction (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.cc1099 const HloSendRecvInstruction* send_recv1 = in CheckSameIsHostTransfer()
1100 DynCast<const HloSendRecvInstruction>(instr1); in CheckSameIsHostTransfer()
1101 const HloSendRecvInstruction* send_recv2 = in CheckSameIsHostTransfer()
1102 DynCast<const HloSendRecvInstruction>(instr2); in CheckSameIsHostTransfer()
1120 const HloSendRecvInstruction* sendrecv = in VerifySendsAndRecvs()
1121 DynCast<const HloSendRecvInstruction>(instruction); in VerifySendsAndRecvs()
Dhlo_instructions.h209 class HloSendRecvInstruction : public HloInstruction {
223 explicit HloSendRecvInstruction(HloOpcode opcode, const Shape& shape,
240 class HloSendInstruction : public HloSendRecvInstruction {
252 class HloSendDoneInstruction : public HloSendRecvInstruction {
264 class HloRecvInstruction : public HloSendRecvInstruction {
276 class HloRecvDoneInstruction : public HloSendRecvInstruction {
Dhlo_instructions.cc363 HloSendRecvInstruction::HloSendRecvInstruction(HloOpcode opcode, in HloSendRecvInstruction() function in xla::HloSendRecvInstruction
371 HloInstructionProto HloSendRecvInstruction::ToProto() const { in ToProto()
378 std::vector<string> HloSendRecvInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()
388 bool HloSendRecvInstruction::IdenticalSlowPath( in IdenticalSlowPath()
400 : HloSendRecvInstruction( in HloSendInstruction()
420 : HloSendRecvInstruction(HloOpcode::kSendDone, ShapeUtil::MakeTokenShape(), in HloSendDoneInstruction()
439 : HloSendRecvInstruction( in HloRecvInstruction()
458 : HloSendRecvInstruction( in HloRecvDoneInstruction()
Dhlo_module_group_metadata.cc188 const HloSendRecvInstruction* send_recv_instr = in IsChannelInstruction()
189 DynCast<HloSendRecvInstruction>(instruction); in IsChannelInstruction()
Dlayout_assignment.cc384 const HloSendRecvInstruction* send_recv_instr = in IsHostSendRecv()
385 DynCast<HloSendRecvInstruction>(instruction); in IsHostSendRecv()
394 const HloSendRecvInstruction* send_recv_instr = in BuildHostChannelConstraints()
395 DynCast<HloSendRecvInstruction>(instruction); in BuildHostChannelConstraints()
Dhlo_memory_scheduler.cc218 DynCast<HloSendRecvInstruction>(instruction)->is_host_transfer()) { in BytesFreedIfScheduled()
Dhlo_instruction.cc3231 return Cast<HloSendRecvInstruction>(this)->channel_id(); in channel_id()