Searched defs:ReplicaGroups (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/dtensor/mlir/ | ||
| D | group_assignment.cc | 64 GroupAssignment::ReplicaGroups::ReplicaGroups( in ReplicaGroups() function in tensorflow::dtensor::GroupAssignment::ReplicaGroups |
| /external/tensorflow/tensorflow/compiler/xla/service/ | ||
| D | hlo_matchers.h | 523 inline ::testing::Matcher<const ::xla::HloInstruction*> ReplicaGroups( in ReplicaGroups() function |