Searched defs:replica_groups (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | collective_ops_utils.cc | 47 int64 device_ordinal, absl::Span<const ReplicaGroup> replica_groups, in GetParticipatingReplicas()
|
| D | bfloat16_normalization_test.cc | 286 std::vector<ReplicaGroup> replica_groups(1); in TEST_F() local
315 std::vector<ReplicaGroup> replica_groups(1); in TEST_F() local
|
| D | hlo_verifier_test.cc | 853 string ReplicaGroupsStr(std::vector<std::vector<int64>> replica_groups) { in ReplicaGroupsStr()
863 std::vector<std::vector<int64>> replica_groups) { in MakeAllReduceComputation()
911 std::vector<std::vector<int64>> replica_groups) { in MakeAllToAllComputation()
|
| D | hlo_parser.cc | 413 std::vector<ReplicaGroup> replica_groups; in CreateReplicaGroups() local
865 std::vector<ReplicaGroup> replica_groups; in ParseInstructionRhs() local
887 std::vector<ReplicaGroup> replica_groups; in ParseInstructionRhs() local
2074 std::vector<ReplicaGroup>* replica_groups) { in ParseReplicaGroupsOnly()
4167 std::vector<ReplicaGroup> replica_groups; in ParseReplicaGroupsOnly() local
|
| D | hlo.proto | 177 repeated ReplicaGroup replica_groups = 49; field
|
| D | hlo_instruction.cc | 913 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout, in CreateAllReduce()
922 const std::vector<ReplicaGroup>& replica_groups, in CreateAllToAll()
3394 string ReplicaGroupsToString(const std::vector<ReplicaGroup>& replica_groups) { in ReplicaGroupsToString()
3757 const std::vector<ReplicaGroup>& HloInstruction::replica_groups() const { in replica_groups() function in xla::HloInstruction
|
| D | hlo_instructions.h | 311 const std::vector<ReplicaGroup>& replica_groups() const { in replica_groups() function
|
| D | hlo_instructions.cc | 515 const std::vector<ReplicaGroup>& replica_groups, in HloCollectiveInstruction()
556 const std::vector<ReplicaGroup>& replica_groups, bool constrain_layout, in HloAllReduceInstruction()
610 const std::vector<ReplicaGroup>& replica_groups, in HloAllToAllInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1008 def AllReduce(self, operand, computation, replica_groups=None): argument
1030 replica_groups=None): argument
1056 def CrossReplicaSum(self, operand, replica_groups=None): argument
1901 def _get_replica_groups_protos(replica_groups): argument
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | collective_ops_test.cc | 44 const Shape& shape, std::vector<std::vector<int64>> replica_groups, in MakeCrsModule()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2084 XlaOp operand, absl::Span<const ReplicaGroup> replica_groups) { in CrossReplicaSum()
2114 absl::Span<const ReplicaGroup> replica_groups, in AllReduce()
2191 const std::vector<ReplicaGroup>& replica_groups) { in AllToAll()
3302 absl::Span<const ReplicaGroup> replica_groups) { in CrossReplicaSum()
3307 absl::Span<const ReplicaGroup> replica_groups, in AllReduce()
3316 const std::vector<ReplicaGroup>& replica_groups) { in AllToAll()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 487 auto replica_groups = Convert_replica_groups(op.replica_groups()); in ExportXlaOp() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 1464 std::string replica_groups = ReplicaGroupsToString(crs->replica_groups()); in HandleAllReduceMultipleReplica() local
|