Skip to main content

DispatchIffFn

Type Alias DispatchIffFn 

Source
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.