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