Searched defs:ReplicaId (Results 1 – 3 of 3) sorted by relevance
2190 ## ReplicaId section in Operation Semantics
530 XLA_TEST_F(CollectiveOpsTest, ReplicaId) { in XLA_TEST_F() argument
3018 XlaOp XlaBuilder::ReplicaId() { in ReplicaId() function in xla::XlaBuilder4700 XlaOp ReplicaId(XlaBuilder* builder) { return builder->ReplicaId(); } in ReplicaId() function