Searched defs:constrain_layout (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo.proto | 210 bool constrain_layout = 56; field
|
| D | hlo_parser.cc | 1201 optional<bool> constrain_layout; in ParseInstructionRhs() local
1230 optional<bool> constrain_layout; in ParseInstructionRhs() local
1263 optional<bool> constrain_layout; in ParseInstructionRhs() local
|
| D | hlo_instructions.cc | 573 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout, in HloCollectiveInstruction()
620 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout, in HloAllGatherInstruction()
668 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout, in HloAllReduceInstruction()
724 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout, in HloAllToAllInstruction()
|
| D | hlo_instructions.h | 366 bool constrain_layout() const { return constrain_layout_; } in constrain_layout() function
|
| D | hlo_instruction.cc | 1045 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout, in CreateAllGather()
1055 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout, in CreateAllReduce()
1064 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout, in CreateAllToAll()
|