Searched refs:XlaReplicaId (Results 1 – 4 of 4) sorted by relevance
733 // XlaReplicaId op.
5155 %0 = "tf.XlaReplicaId"() : () -> tensor<i32>
50406 func XlaReplicaId(scope *Scope) (id tf.Output) { func
17964 def TF_XlaReplicaIdOp : TF_Op<"XlaReplicaId", [NoSideEffect, TF_NoConstantFold]> {