Searched defs:CrossReplicaSum (Results 1 – 3 of 3) sorted by relevance
933 ## CrossReplicaSum section in Operation Semantics
2906 XlaOp XlaBuilder::CrossReplicaSum( in CrossReplicaSum() function in xla::XlaBuilder4664 XlaOp CrossReplicaSum(const XlaOp operand, in CrossReplicaSum() function
8011 func CrossReplicaSum(scope *Scope, input tf.Output, group_assignment tf.Output) (output tf.Output) { func