Skip to main content

dispatch_sqrt_f32

mlx_native::ops::sqrt_elementwise

Function dispatch_sqrt_f32 

Source 
pub fn dispatch_sqrt_f32(
    encoder: &mut CommandEncoder,
    registry: &mut KernelRegistry,
    device: &DeviceRef,
    input: &MlxBuffer,
    output: &MlxBuffer,
    params: &MlxBuffer,
) -> Result<()>