| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_dce.cc | 57 !maybe_collective_op->constrain_layout())))) { in RunOnComputation()
|
| D | hlo_instructions.cc | 573 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout, in HloCollectiveInstruction() argument
577 constrain_layout_(constrain_layout) { in HloCollectiveInstruction()
611 constrain_layout() == casted_other.constrain_layout() && in IdenticalSlowPathIgnoringChannelIdValues()
620 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout, in HloAllGatherInstruction() argument
623 replica_groups, constrain_layout, channel_id),
644 constrain_layout(), channel_id(), use_global_device_ids()); in CloneWithNewOperandsImpl()
668 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout, in HloAllReduceInstruction() argument
671 replica_groups, constrain_layout, channel_id), in HloAllReduceInstruction()
708 constrain_layout() == casted_other.constrain_layout() && in IdenticalSlowPathIgnoringChannelIdValues()
718 shape, new_operands, to_apply(), replica_groups(), constrain_layout(), in CloneWithNewOperandsImpl()
[all …]
|
| D | hlo_query.cc | 128 DynCast<HloAllReduceInstruction>(hlo)->constrain_layout()) { in ContainsLayoutConstrainedAllReduce()
|
| D | all_to_all_decomposer.cc | 44 if (all_to_all->constrain_layout()) { in InstructionMatchesPattern()
|
| D | while_loop_all_reduce_code_motion.cc | 432 old_all_reduce->constrain_layout(), in CreateSinkedAllReduces()
561 if (all_reduce_instruction->constrain_layout()) { in Run()
|
| D | all_gather_decomposer.cc | 133 /*constrain_layout=*/ag->constrain_layout(), ag->channel_id(), in DecomposeAllGather()
|
| D | hlo_instruction.cc | 427 proto.constrain_layout(), channel_id, proto.use_global_device_ids()); in CreateFromProto()
448 /*constrain_layout=*/proto.constrain_layout(), in CreateFromProto()
472 /*constrain_layout=*/proto.constrain_layout(), in CreateFromProto()
550 if (proto.constrain_layout()) { in CreateFromProto()
1045 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout, in CreateAllGather() argument
1048 shape, operand, all_gather_dimension, replica_groups, constrain_layout, in CreateAllGather()
1055 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout, in CreateAllReduce() argument
1058 shape, operands, reduce_computation, replica_groups, constrain_layout, in CreateAllReduce()
1064 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout, in CreateAllToAll() argument
1068 shape, operands, replica_groups, constrain_layout, channel_id, in CreateAllToAll()
[all …]
|
| D | hlo_parser.cc | 1201 optional<bool> constrain_layout; in ParseInstructionRhs() local
1209 &constrain_layout}; in ParseInstructionRhs()
1221 constrain_layout ? *constrain_layout : false, channel_id, in ParseInstructionRhs()
1230 optional<bool> constrain_layout; in ParseInstructionRhs() local
1238 &constrain_layout}; in ParseInstructionRhs()
1250 constrain_layout ? *constrain_layout : false, channel_id, in ParseInstructionRhs()
1263 optional<bool> constrain_layout; in ParseInstructionRhs() local
1265 &constrain_layout}; in ParseInstructionRhs()
1280 constrain_layout ? *constrain_layout : false, channel_id, in ParseInstructionRhs()
|
| D | hlo_instructions.h | 366 bool constrain_layout() const { return constrain_layout_; } in constrain_layout() function
372 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout,
392 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout,
425 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout,
467 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout,
|
| D | hlo_instruction.h | 635 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout,
654 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout,
687 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout,
|
| D | while_loop_all_reduce_code_motion_test.cc | 123 EXPECT_FALSE(moved_all_reduce->constrain_layout()); in TEST_F()
|
| D | hlo.proto | 207 // constrain_layout is true then the 'shape' field must contain a layout, and
210 bool constrain_layout = 56; field
|
| D | hlo_verifier.cc | 1523 if (reference->constrain_layout() != all_reduce->constrain_layout()) { in VerifyLayoutConstrainedAllReduce()
|
| D | hlo_parser_test.cc | 1571 ROOT crs = f32[8]{0} all-reduce(input), replica_groups={}, constrain_layout=true, to_apply=add in CreateTestCases()
1614 …ROOT ag = f32[128,128]{0,1} all-gather(input), replica_groups={}, constrain_layout=true, dimension… in CreateTestCases()
|
| D | layout_assignment.cc | 439 return collective != nullptr && collective->constrain_layout(); in IsLayoutConstrainedCollective()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | schedule_aware_all_gather_cse.cc | 34 if (coll->constrain_layout()) { in MayConsiderAsAllGather()
|
| D | canonicalize_all_gather_for_cse.cc | 85 ag->constrain_layout(), new_channel_id, in RunOnComputation()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | lhlo_ops.mlir | 12 {channel_id = {handle = 1 : i64, type = 0 : i64}, constrain_layout = false,
27 {channel_id = {handle = 1 : i64, type = 0 : i64}, constrain_layout = false,
38 {channel_id = {handle = 1 : i64, type = 0 : i64}, constrain_layout = false,
49 {channel_id = {handle = 1 : i64, type = 0 : i64}, constrain_layout = false,
60 {channel_id = {handle = 1 : i64, type = 0 : i64}, constrain_layout = false,
768 constrain_layout = true,
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | lhlo_ops.td | 548 DefaultValuedAttr<BoolAttr, "false">:$constrain_layout,
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 620 // CHECK-SAME: constrain_layout = false
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1041 op.constrain_layoutAttr(builder.getBoolAttr(collective->constrain_layout())); in SetupCommonCollectiveOpAttributes()
|