Searched refs:ReplicaId (Results 1 – 9 of 9) sorted by relevance
446 LocalOp ComputationBuilder::ReplicaId() { return xla::ReplicaId(&builder_); } in ReplicaId() function in xla::swig::ComputationBuilder
227 LocalOp ReplicaId();
293 %unignore xla::swig::ComputationBuilder::ReplicaId;
1163 def ReplicaId(self): member in ComputationBuilder1169 return self._client.ReplicaId()
611 _ = c.ReplicaId()
482 XlaOp ReplicaId();896 friend XlaOp ReplicaId(XlaBuilder* builder);1610 XlaOp ReplicaId(XlaBuilder* builder);
2236 XlaOp XlaBuilder::ReplicaId() { in ReplicaId() function in xla::XlaBuilder3257 XlaOp ReplicaId(XlaBuilder* builder) { return builder->ReplicaId(); } in ReplicaId() function
1946 ## ReplicaId section in Operation Semantics1949 [`XlaBuilder::ReplicaId`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_builde…1953 <b> `ReplicaId()` </b>1957 program, a `ReplicaId()` call in the program will return a different value on
1431 "ReplicaId", in CreateTestCases()