tensorflow::ops

Function cross_replica_sum

Source
pub fn cross_replica_sum<O0: Into<Output>, O1: Into<Output>>(
    input: O0,
    group_assignment: O1,
    scope: &mut Scope,
) -> Result<Operation>
Expand description

Shorthand for CrossReplicaSum::new().build(input, group_assignment, scope).