Searched defs:CrossReplicaSum (Results 1 – 5 of 5) sorted by relevance
781 ## CrossReplicaSum section in Operation Semantics
2120 XlaOp XlaBuilder::CrossReplicaSum( in CrossReplicaSum() function in xla::XlaBuilder2134 XlaOp XlaBuilder::CrossReplicaSum( in CrossReplicaSum() function in xla::XlaBuilder3232 XlaOp CrossReplicaSum(const XlaOp& operand, in CrossReplicaSum() function3237 XlaOp CrossReplicaSum(const XlaOp& operand, const XlaComputation& computation, in CrossReplicaSum() function
510 LocalOp ComputationBuilder::CrossReplicaSum( in CrossReplicaSum() function in xla::swig::ComputationBuilder
1241 def CrossReplicaSum(self, operand, replica_groups=None): member in ComputationBuilder
20923 func CrossReplicaSum(scope *Scope, input tf.Output, group_assignment tf.Output) (output tf.Output) { func