Searched defs:ReplicaId (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ | ||
| D | operation_semantics.md | 1946 ## ReplicaId section in Operation Semantics |
| /external/tensorflow/tensorflow/compiler/xla/python/ | ||
| D | local_computation_builder.cc | 446 LocalOp ComputationBuilder::ReplicaId() { return xla::ReplicaId(&builder_); } in ReplicaId() function in xla::swig::ComputationBuilder |
| D | xla_client.py | 1163 def ReplicaId(self): member in ComputationBuilder |
| /external/tensorflow/tensorflow/compiler/xla/client/ | ||
| D | xla_builder.cc | 2236 XlaOp XlaBuilder::ReplicaId() { in ReplicaId() function in xla::XlaBuilder |