Home
last modified time | relevance | path

Searched refs:is_host_transfer (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_group_util.cc99 !DynCast<HloRecvDoneInstruction>(instruction)->is_host_transfer()) { in GlobalPredecessors()
106 !DynCast<HloSendInstruction>(instruction)->is_host_transfer()) { in GlobalPredecessors()
177 !DynCast<HloRecvInstruction>(instruction)->is_host_transfer()) { in GlobalSuccessors()
186 !DynCast<HloSendInstruction>(instruction)->is_host_transfer()) { in GlobalSuccessors()
Dhlo_query.cc156 if (send->is_host_transfer() && send->operand(0)->shape().IsTuple()) { in HasX64TransformedHostTransfer()
161 if (recv->is_host_transfer() && in HasX64TransformedHostTransfer()
Dhlo_instructions.cc462 bool is_host_transfer) in HloSendRecvInstruction() argument
464 is_host_transfer_(is_host_transfer) {} in HloSendRecvInstruction()
476 if (is_host_transfer()) { in ExtraAttributesToStringImpl()
493 bool is_host_transfer) in HloSendInstruction() argument
499 channel_id, is_host_transfer) {
509 new_operands[0], new_operands[1], *channel_id(), is_host_transfer()); in CloneWithNewOperandsImpl()
513 bool is_host_transfer) in HloSendDoneInstruction() argument
516 is_host_transfer) { in HloSendDoneInstruction()
526 Cast<HloSendInstruction>(new_operands[0]), is_host_transfer()); in CloneWithNewOperandsImpl()
532 bool is_host_transfer) in HloRecvInstruction() argument
[all …]
Dhlo_instructions.h279 bool is_host_transfer() const { return is_host_transfer_; } in is_host_transfer() function
286 int64 channel_id, bool is_host_transfer);
302 int64 channel_id, bool is_host_transfer);
314 bool is_host_transfer);
326 int64 channel_id, bool is_host_transfer);
338 bool is_host_transfer);
Dhlo_instruction.cc212 proto.is_host_transfer()); in CreateFromProto()
215 instruction = CreateSendDone(operands(0), proto.is_host_transfer()); in CreateFromProto()
219 proto.channel_id(), proto.is_host_transfer()); in CreateFromProto()
222 instruction = CreateRecvDone(operands(0), proto.is_host_transfer()); in CreateFromProto()
1124 bool is_host_transfer) { in CreateSend() argument
1126 is_host_transfer); in CreateSend()
1130 HloInstruction* operand, bool is_host_transfer) { in CreateSendDone() argument
1135 is_host_transfer); in CreateSendDone()
1140 bool is_host_transfer) { in CreateRecv() argument
1142 is_host_transfer); in CreateRecv()
[all …]
Dhlo_parser.cc1457 optional<bool> is_host_transfer = false; in ParseInstructionRhs() local
1460 &is_host_transfer}; in ParseInstructionRhs()
1467 shape.tuple_shapes(0), operands[0], *channel_id, *is_host_transfer)); in ParseInstructionRhs()
1473 optional<bool> is_host_transfer = false; in ParseInstructionRhs() local
1476 &is_host_transfer}; in ParseInstructionRhs()
1488 HloInstruction::CreateRecvDone(operands[0], *is_host_transfer)); in ParseInstructionRhs()
1494 optional<bool> is_host_transfer = false; in ParseInstructionRhs() local
1497 &is_host_transfer}; in ParseInstructionRhs()
1503 operands[0], operands[1], *channel_id, *is_host_transfer)); in ParseInstructionRhs()
1509 optional<bool> is_host_transfer = false; in ParseInstructionRhs() local
[all …]
Dhlo_instruction.h752 bool is_host_transfer = false);
757 HloInstruction* operand, bool is_host_transfer = false);
766 bool is_host_transfer = false);
771 HloInstruction* operand, bool is_host_transfer = false);
Dhlo_parser_test.cc353 %recv = (f32[], u32[], token[]) recv(token[] %token0), channel_id=15, is_host_transfer=true in CreateTestCases()
354 …e = (f32[], token[]) recv-done((f32[], u32[], token[]) %recv), channel_id=15, is_host_transfer=true in CreateTestCases()
356 …f32[], u32[], token[]) send(f32[] %constant, token[] %token0), channel_id=16, is_host_transfer=true in CreateTestCases()
357 …%send-done = token[] send-done((f32[], u32[], token[]) %send), channel_id=16, is_host_transfer=true in CreateTestCases()
Dhlo_verifier.cc1436 if (send_recv1->is_host_transfer() != send_recv2->is_host_transfer()) { in CheckSameIsHostTransfer()
1594 if (sendrecv->is_host_transfer()) { in VerifyChannels()
Dhlo_module_group_metadata.cc208 return !send_recv_instr->is_host_transfer(); in IsChannelInstruction()
Dlayout_assignment.cc396 return send_recv_instr != nullptr && send_recv_instr->is_host_transfer(); in IsHostSendRecv()
406 if (send_recv_instr == nullptr || !send_recv_instr->is_host_transfer()) { in BuildHostChannelConstraints()
Dhlo.proto189 bool is_host_transfer = 47; field
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dexport.mlir611 is_host_transfer = true
618 …ECV:%.*]] = (s32[3,4], u32[], token[]) recv(token[] [[TOKEN]]), channel_id=5, is_host_transfer=true
620 …[3,4], token[]) recv-done((s32[3,4], u32[], token[]) [[RECV]]), channel_id=5, is_host_transfer=true
631 is_host_transfer = false
842 is_host_transfer = true
850 …,4], u32[], token[]) send(s32[3,4] [[ARG]], token[] [[TOKEN]]), channel_id=5, is_host_transfer=true
852 …-SAME: token[] send-done((s32[3,4], u32[], token[]) [[SEND]]), channel_id=5, is_host_transfer=true
863 is_host_transfer = false
1104 …%0 = "mhlo.send"(%arg, %token) {channel_id = {handle = 1 : i64, type = 2 : i64}, is_host_transfer …
1105 …%1 = "mhlo.recv"(%0) {channel_id = {handle = 2 : i64, type = 3 : i64}, is_host_transfer = true, mh…
[all …]
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-communication.mlir24 // CHECK-SAME: is_host_transfer = true
31 // CHECK-SAME: is_host_transfer = true
40 // CHECK-SAME: is_host_transfer = true
57 // CHECK-SAME: is_host_transfer = true
171 // CHECK-SAME: is_host_transfer = true
188 // CHECK-SAME: is_host_transfer = true
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops.td462 DefaultValuedAttr<BoolAttr, "false">:$is_host_transfer
487 DefaultValuedAttr<BoolAttr, "false">:$is_host_transfer
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc843 if (op.is_host_transfer()) { in ExportXlaOp()
989 if (op.is_host_transfer()) { in ExportXlaOp()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dops.mlir614 is_host_transfer = true
628 is_host_transfer = true