Home
last modified time | relevance | path

Searched refs:ReplicaGroup (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_data.i634 %typemap(in) absl::Span<const ReplicaGroup >
635 (std::vector<ReplicaGroup > temps) {
644 ReplicaGroup rgrp;
Dlocal_computation_builder.h257 absl::Span<const ReplicaGroup> replica_groups);
260 absl::Span<const ReplicaGroup> replica_groups);
Dlocal_computation_builder.cc500 int64 split_count, absl::Span<const ReplicaGroup> replica_groups) { in AllToAll()
501 std::vector<ReplicaGroup> rg; in AllToAll()
511 const LocalOp& operand, absl::Span<const ReplicaGroup> replica_groups) { in CrossReplicaSum()
Dxla_client.py1973 class ReplicaGroup(object): class
1982 replica_group_proto = ReplicaGroup()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h467 absl::Span<const ReplicaGroup> replica_groups = {});
471 absl::Span<const ReplicaGroup> replica_groups = {},
476 const std::vector<ReplicaGroup>& replica_groups);
885 absl::Span<const ReplicaGroup> replica_groups);
888 absl::Span<const ReplicaGroup> replica_groups,
892 const std::vector<ReplicaGroup>& replica_groups);
1568 absl::Span<const ReplicaGroup> replica_groups = {});
1589 absl::Span<const ReplicaGroup> replica_groups = {},
1595 const std::vector<ReplicaGroup>& replica_groups = {});
Dxla_builder.cc2121 const XlaOp& operand, absl::Span<const ReplicaGroup> replica_groups) { in CrossReplicaSum()
2136 absl::Span<const ReplicaGroup> replica_groups, in CrossReplicaSum()
2145 for (const ReplicaGroup& group : replica_groups) { in CrossReplicaSum()
2161 const std::vector<ReplicaGroup>& replica_groups) { in AllToAll()
2197 for (const ReplicaGroup& group : replica_groups) { in AllToAll()
3233 absl::Span<const ReplicaGroup> replica_groups) { in CrossReplicaSum()
3238 absl::Span<const ReplicaGroup> replica_groups, in CrossReplicaSum()
3246 const std::vector<ReplicaGroup>& replica_groups) { in AllToAll()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.h290 const std::vector<ReplicaGroup>& replica_groups() const { in replica_groups()
298 const std::vector<ReplicaGroup>& replica_groups);
309 std::vector<ReplicaGroup> replica_groups_;
317 const std::vector<ReplicaGroup>& replica_groups,
361 const std::vector<ReplicaGroup>& replica_groups);
Dar_crs_combiner_test.cc329 void CompareReplicaGroups(const std::vector<ReplicaGroup>& groups_before, in CompareReplicaGroups()
330 const std::vector<ReplicaGroup>& groups_after) { in CompareReplicaGroups()
Dhlo_instruction.h490 const std::vector<ReplicaGroup>& replica_groups,
508 const std::vector<ReplicaGroup>& replica_groups);
1522 const std::vector<ReplicaGroup>& replica_groups() const;
Dhlo_instructions.cc479 const std::vector<ReplicaGroup>& replica_groups) in HloCollectiveInstruction()
497 for (const ReplicaGroup& group : replica_groups()) { in ExtraAttributesToStringImpl()
513 [](const ReplicaGroup& a, const ReplicaGroup& b) { in IdenticalSlowPath()
521 const std::vector<ReplicaGroup>& replica_groups, absl::string_view barrier, in HloAllReduceInstruction()
589 const std::vector<ReplicaGroup>& replica_groups) in HloAllToAllInstruction()
Dhlo_instruction.cc386 std::vector<ReplicaGroup>(proto.replica_groups().begin(), in CreateFromProto()
396 std::vector<ReplicaGroup>(proto.replica_groups().begin(), in CreateFromProto()
835 const std::vector<ReplicaGroup>& replica_groups, absl::string_view barrier, in CreateAllReduce()
844 const std::vector<ReplicaGroup>& replica_groups) { in CreateAllToAll()
3412 const std::vector<ReplicaGroup>& HloInstruction::replica_groups() const { in replica_groups()
Dhlo_parser.cc393 std::vector<ReplicaGroup> CreateReplicaGroups( in CreateReplicaGroups()
395 std::vector<ReplicaGroup> replica_groups; in CreateReplicaGroups()
398 ReplicaGroup group; in CreateReplicaGroups()
833 std::vector<ReplicaGroup> replica_groups; in ParseInstructionRhs()
850 std::vector<ReplicaGroup> replica_groups; in ParseInstructionRhs()
Dhlo.proto173 repeated ReplicaGroup replica_groups = 49;
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto608 message ReplicaGroup { message
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md48 - `replica_groups`: each ReplicaGroup contains a list of replica id. If empty,
84 | `replica_groups` | `ReplicaGroup` vector | each group contains a list of |