Function tensorflow::ops::all_to_all
source · pub fn all_to_all<O0: Into<Output>, O1: Into<Output>>(
input: O0,
group_assignment: O1,
scope: &mut Scope
) -> Result<Operation>
Expand description
Shorthand for AllToAll::new().build(input, group_assignment, scope)
.