Searched defs:constrain_layout (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_instruction.cc | 1260 bool constrain_layout, const std::optional<int64_t>& channel_id, in CreateAllGather()
1271 bool constrain_layout, const std::optional<int64_t>& channel_id, in CreateAllGatherStart()
1281 absl::Span<const ReplicaGroup> replica_groups, bool constrain_layout, in CreateAllReduce()
1292 absl::Span<const ReplicaGroup> replica_groups, bool constrain_layout, in CreateReduceScatter()
1304 absl::Span<const ReplicaGroup> replica_groups, bool constrain_layout, in CreateAllReduceStart()
1313 absl::Span<const ReplicaGroup> replica_groups, bool constrain_layout, in CreateAllToAll()
|
| H A D | hlo_instructions.cc | 740 absl::Span<const ReplicaGroup> replica_groups, bool constrain_layout, in HloCollectiveInstruction()
788 absl::Span<const ReplicaGroup> replica_groups, bool constrain_layout, in HloAllGatherInstruction()
837 absl::Span<const ReplicaGroup> replica_groups, bool constrain_layout, in HloAllReduceInstructionBase()
899 absl::Span<const ReplicaGroup> replica_groups, bool constrain_layout, in HloReduceScatterInstruction()
944 absl::Span<const ReplicaGroup> replica_groups, bool constrain_layout, in HloAllToAllInstruction()
|
| H A D | hlo_parser.cc | 1432 optional<bool> constrain_layout; in CreateInstruction() local
1469 optional<bool> constrain_layout; in CreateInstruction() local
1519 optional<bool> constrain_layout; in CreateInstruction() local
|
| H A D | hlo.proto | 265 bool constrain_layout = 56; field
|
| H A D | hlo_instructions.h | 551 bool constrain_layout() const { return constrain_layout_; } in constrain_layout() function
|