tensorflow::ops

Function nccl_all_reduce

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

Shorthand for NcclAllReduce::new().build(input, scope).