Searched defs:CrossReplicaSum (Results 1 – 4 of 4) sorted by relevance
848 ## CrossReplicaSum section in Operation Semantics
1056 def CrossReplicaSum(self, operand, replica_groups=None): member in ComputationBuilder
2083 XlaOp XlaBuilder::CrossReplicaSum( in CrossReplicaSum() function in xla::XlaBuilder3301 XlaOp CrossReplicaSum(const XlaOp operand, in CrossReplicaSum() function
44984 func CrossReplicaSum(scope *Scope, input tf.Output, group_assignment tf.Output) (output tf.Output) { func