| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | cross_replica_ops.cc | 34 std::vector<xla::ReplicaGroup> replica_groups; in Convert() local
59 std::vector<xla::ReplicaGroup> replica_groups = in Compile() local
81 std::vector<xla::ReplicaGroup> replica_groups = in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | collective_ops_utils.cc | 55 absl::Span<const ReplicaGroup> replica_groups) { in GetParticipatingReplicas()
80 int total_replica_count, absl::Span<const ReplicaGroup> replica_groups) { in GetParticipatingDevices()
|
| D | collective_ops_utils_test.cc | 36 std::vector<ReplicaGroup> replica_groups(3); in TEST() local
76 std::vector<ReplicaGroup> replica_groups(2); in TEST() local
|
| D | bfloat16_normalization_test.cc | 287 std::vector<ReplicaGroup> replica_groups(1); in TEST_F() local
316 std::vector<ReplicaGroup> replica_groups(1); in TEST_F() local
|
| D | hlo_verifier_test.cc | 872 string ReplicaGroupsStr(std::vector<std::vector<int64>> replica_groups) { in ReplicaGroupsStr()
881 int64 ReplicaCount(const std::vector<std::vector<int64>>& replica_groups) { in ReplicaCount()
890 std::vector<std::vector<int64>> replica_groups, in MakeAllReduceComputation()
963 std::vector<std::vector<int64>> replica_groups) { in MakeAllToAllComputation()
|
| D | all_reduce_combiner.cc | 156 std::vector<ReplicaGroup> replica_groups; member
|
| D | all_reduce_combiner_test.cc | 103 std::vector<ReplicaGroup> replica_groups(groups.size()); in CreateReplicaGroups() local
|
| D | hlo_parser.cc | 565 std::vector<ReplicaGroup> replica_groups; in CreateReplicaGroups() local
1215 std::vector<ReplicaGroup> replica_groups; in ParseInstructionRhs() local
1244 std::vector<ReplicaGroup> replica_groups; in ParseInstructionRhs() local
1270 std::vector<ReplicaGroup> replica_groups; in ParseInstructionRhs() local
2887 std::vector<ReplicaGroup>* replica_groups) { in ParseReplicaGroupsOnly()
5138 std::vector<ReplicaGroup> replica_groups; in ParseReplicaGroupsOnly() local
|
| D | ar_crs_combiner.cc | 97 auto replica_groups = all_reduce->replica_groups(); in HasCombinableReplicaGroup() local
|
| 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()
3733 string ReplicaGroupsToString(const std::vector<ReplicaGroup>& replica_groups) { in ReplicaGroupsToString()
4127 const std::vector<ReplicaGroup>& HloInstruction::replica_groups() const { in replica_groups() function in xla::HloInstruction
|
| D | hlo.proto | 177 repeated ReplicaGroup replica_groups = 49; field
|
| 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 | 349 const std::vector<ReplicaGroup>& replica_groups() const { in replica_groups() function
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_exporter.cc | 100 std::vector<ReplicaGroup> replica_groups(type.getDimSize(0)); in ConvertReplicaGroups() local
|
| D | hlo_function_importer.cc | 953 const std::vector<ReplicaGroup>& replica_groups, mlir::Builder* builder) { in ConvertReplicaGroups()
|
| D | mlir_hlo_to_hlo.cc | 579 auto replica_groups = Convert_replica_groups(op.replica_groups()); in ExportXlaOp() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_collective_thunk.h | 71 std::vector<ReplicaGroup> replica_groups; member
|
| /external/tensorflow/tensorflow/core/profiler/protobuf/ |
| D | pod_viewer.proto | 29 repeated ReplicaGroup replica_groups = 5; field
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 677 def make_replica_groups(replica_groups): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2716 absl::Span<const ReplicaGroup> replica_groups, in AllGather()
2752 XlaOp operand, absl::Span<const ReplicaGroup> replica_groups) { in CrossReplicaSum()
2782 absl::Span<const ReplicaGroup> replica_groups, in AllReduce()
2859 const std::vector<ReplicaGroup>& replica_groups, in AllToAll()
2873 int64 split_count, const std::vector<ReplicaGroup>& replica_groups) { in AllToAllArray()
2925 const std::vector<ReplicaGroup>& replica_groups, in AllToAllTuple()
4655 absl::Span<const ReplicaGroup> replica_groups, in AllGather()
4665 absl::Span<const ReplicaGroup> replica_groups) { in CrossReplicaSum()
4670 absl::Span<const ReplicaGroup> replica_groups, in AllReduce()
4679 const std::vector<ReplicaGroup>& replica_groups, in AllToAll()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | collective_ops_test.cc | 43 const Shape& shape, std::vector<std::vector<int64>> replica_groups, in MakeCrsModule()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 1135 std::string replica_groups = ReplicaGroupsToString(crs->replica_groups()); in HandleAllReduceMultipleReplica() local
1215 std::string replica_groups = in HandleAllToAll() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5266 auto replica_groups = in matchAndRewrite() local
|