Function tensorflow::ops::collective_all_to_all_v2
source · pub fn collective_all_to_all_v2<O0: Into<Output>, O1: Into<Output>, O2: Into<Output>, O3: Into<Output>, O4: Into<Output>>(
input: O0,
group_size: O1,
group_key: O2,
instance_key: O3,
ordering_token: O4,
scope: &mut Scope
) -> Result<Operation>
Expand description
Shorthand for CollectiveAllToAllV2::new().build(input, group_size, group_key, instance_key, ordering_token, scope)
.