| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_group_util.cc | 100 !DynCast<HloRecvDoneInstruction>(instruction)->is_host_transfer()) { in GlobalPredecessors()
106 !DynCast<HloSendInstruction>(instruction)->is_host_transfer()) { in GlobalPredecessors()
178 !DynCast<HloRecvInstruction>(instruction)->is_host_transfer()) { in GlobalSuccessors()
186 !DynCast<HloSendInstruction>(instruction)->is_host_transfer()) { in GlobalSuccessors()
|
| D | hlo_instructions.cc | 366 bool is_host_transfer) in HloSendRecvInstruction() argument
369 is_host_transfer_(is_host_transfer) {} in HloSendRecvInstruction()
382 if (is_host_transfer()) { in ExtraAttributesToStringImpl()
399 bool is_host_transfer) in HloSendInstruction() argument
405 channel_id, is_host_transfer) {
415 new_operands[0], new_operands[1], channel_id(), is_host_transfer()); in CloneWithNewOperandsImpl()
419 bool is_host_transfer) in HloSendDoneInstruction() argument
422 is_host_transfer) { in HloSendDoneInstruction()
432 Cast<HloSendInstruction>(new_operands[0]), is_host_transfer()); in CloneWithNewOperandsImpl()
438 bool is_host_transfer) in HloRecvInstruction() argument
[all …]
|
| D | hlo_instructions.h | 217 bool is_host_transfer() const { return is_host_transfer_; } in is_host_transfer() function
224 int64 channel_id, bool is_host_transfer);
243 int64 channel_id, bool is_host_transfer);
255 bool is_host_transfer);
267 int64 channel_id, bool is_host_transfer);
279 bool is_host_transfer);
|
| D | hlo_instruction.cc | 190 proto.is_host_transfer()); in CreateFromProto()
193 instruction = CreateSendDone(operands(0), proto.is_host_transfer()); in CreateFromProto()
197 proto.channel_id(), proto.is_host_transfer()); in CreateFromProto()
200 instruction = CreateRecvDone(operands(0), proto.is_host_transfer()); in CreateFromProto()
878 bool is_host_transfer) { in CreateSend() argument
880 is_host_transfer); in CreateSend()
884 HloInstruction* operand, bool is_host_transfer) { in CreateSendDone() argument
889 is_host_transfer); in CreateSendDone()
894 bool is_host_transfer) { in CreateRecv() argument
896 is_host_transfer); in CreateRecv()
[all …]
|
| D | hlo_instruction.h | 553 bool is_host_transfer = false);
558 HloInstruction* operand, bool is_host_transfer = false);
567 bool is_host_transfer = false);
572 HloInstruction* operand, bool is_host_transfer = false);
|
| D | hlo_parser.cc | 960 optional<bool> is_host_transfer = false; in ParseInstructionRhs() local
963 &is_host_transfer}; in ParseInstructionRhs()
970 shape.tuple_shapes(0), operands[0], *channel_id, *is_host_transfer)); in ParseInstructionRhs()
976 optional<bool> is_host_transfer = false; in ParseInstructionRhs() local
979 &is_host_transfer}; in ParseInstructionRhs()
988 HloInstruction::CreateRecvDone(operands[0], *is_host_transfer)); in ParseInstructionRhs()
994 optional<bool> is_host_transfer = false; in ParseInstructionRhs() local
997 &is_host_transfer}; in ParseInstructionRhs()
1003 operands[0], operands[1], *channel_id, *is_host_transfer)); in ParseInstructionRhs()
1009 optional<bool> is_host_transfer = false; in ParseInstructionRhs() local
[all …]
|
| D | hlo_verifier.cc | 1105 if (send_recv1->is_host_transfer() != send_recv2->is_host_transfer()) { in CheckSameIsHostTransfer()
1122 if (sendrecv->is_host_transfer()) { in VerifySendsAndRecvs()
|
| D | hlo_parser_test.cc | 327 %recv = (f32[], u32[], token[]) recv(token[] %token0), channel_id=15, is_host_transfer=true in CreateTestCases()
328 …e = (f32[], token[]) recv-done((f32[], u32[], token[]) %recv), channel_id=15, is_host_transfer=true in CreateTestCases()
330 …f32[], u32[], token[]) send(f32[] %constant, token[] %token0), channel_id=16, is_host_transfer=true in CreateTestCases()
331 …%send-done = token[] send-done((f32[], u32[], token[]) %send), channel_id=16, is_host_transfer=true in CreateTestCases()
|
| D | hlo_memory_scheduler.cc | 218 DynCast<HloSendRecvInstruction>(instruction)->is_host_transfer()) { in BytesFreedIfScheduled()
|
| D | hlo_module_group_metadata.cc | 191 return !send_recv_instr->is_host_transfer(); in IsChannelInstruction()
|
| D | hlo.proto | 180 bool is_host_transfer = 47; field
|
| D | layout_assignment.cc | 386 return send_recv_instr != nullptr && send_recv_instr->is_host_transfer(); in IsHostSendRecv()
396 if (send_recv_instr == nullptr || !send_recv_instr->is_host_transfer()) { in BuildHostChannelConstraints()
|