Home
last modified time | relevance | path

Searched defs:constrain_layout (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instruction.cc1163 bool constrain_layout, const absl::optional<int64>& channel_id, in CreateAllGather()
1174 bool constrain_layout, const absl::optional<int64>& channel_id, in CreateAllGatherStart()
1184 absl::Span<const ReplicaGroup> replica_groups, bool constrain_layout, in CreateAllReduce()
1195 absl::Span<const ReplicaGroup> replica_groups, bool constrain_layout, in CreateReduceScatter()
1207 absl::Span<const ReplicaGroup> replica_groups, bool constrain_layout, in CreateAllReduceStart()
1216 absl::Span<const ReplicaGroup> replica_groups, bool constrain_layout, in CreateAllToAll()
Dhlo.proto245 bool constrain_layout = 56; field
Dhlo_instructions.cc578 absl::Span<const ReplicaGroup> replica_groups, bool constrain_layout, in HloCollectiveInstruction()
626 absl::Span<const ReplicaGroup> replica_groups, bool constrain_layout, in HloAllGatherInstruction()
675 absl::Span<const ReplicaGroup> replica_groups, bool constrain_layout, in HloAllReduceInstructionBase()
736 absl::Span<const ReplicaGroup> replica_groups, bool constrain_layout, in HloReduceScatterInstruction()
780 absl::Span<const ReplicaGroup> replica_groups, bool constrain_layout, in HloAllToAllInstruction()
Dhlo_instructions.h365 bool constrain_layout() const { return constrain_layout_; } in constrain_layout() function
Dhlo_parser.cc1262 optional<bool> constrain_layout; in ParseInstructionRhs() local
1301 optional<bool> constrain_layout; in ParseInstructionRhs() local
1354 optional<bool> constrain_layout; in ParseInstructionRhs() local