Searched defs:ReplicaId (Results 1 – 6 of 6) sorted by relevance
71 absl::flat_hash_map<GroupAssignment::ReplicaId, GroupAssignment::DeviceId> in CreateGroupAssignment()
53 absl::flat_hash_map<ReplicaId, DeviceId> map) in ReplicaToDeviceMap()
2312 ## ReplicaId section in Operation Semantics
2088 struct ReplicaId { struct2092 static ReplicaId Handler() { return ReplicaId(); } in Handler()2116 static bool ReplicaId(runtime::KernelContext* ctx, void** args, void** attrs) { in ReplicaId() function
583 XLA_TEST_F(CollectiveOpsTest, ReplicaId) { in XLA_TEST_F() argument
3248 XlaOp XlaBuilder::ReplicaId() { in ReplicaId() function in xla::XlaBuilder4720 XlaOp ReplicaId(XlaBuilder* builder) { return builder->ReplicaId(); } in ReplicaId() function