pub type DispatchIffFn = fn(cond: &DeviceTensor, then_value: &DeviceTensor, else_value: &DeviceTensor, cond_strides: &[isize], then_strides: &[isize], else_strides: &[isize], output: &DeviceTensor, output_shape: &[usize], output_strides: &[isize]) -> TractResult<()>;Expand description
Dispatch function for the iff (select) kernel. Args: cond, then, else tensors with pre-computed broadcast strides, output tensor, output shape and strides. All strides are padded to IFF_MAX_RANK.