Home
last modified time | relevance | path

Searched refs:ReplicaGroupsToString (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_instruction.h2419 std::string ReplicaGroupsToString(
H A Dhlo_instructions.cc763 StrCat("replica_groups=", ReplicaGroupsToString(replica_groups()))); in ExtraAttributesToStringImpl()
H A Dhlo_instruction.cc4172 std::string ReplicaGroupsToString( in ReplicaGroupsToString() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc1209 std::string replica_groups = ReplicaGroupsToString(crs->replica_groups()); in HandleAllReduceMultipleReplica()
1295 ReplicaGroupsToString(instruction->replica_groups()); in HandleAllToAll()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Ddot_handler.cc745 << ", replica groups: " << ReplicaGroupsToString(groups); in GetLoopReplicaGroups()