Skip to main content

dispatch_exp_backward_f32

mlx_native::ops::exp_elementwise

Function dispatch_exp_backward_f32 

Source 
pub fn dispatch_exp_backward_f32(
    encoder: &mut CommandEncoder,
    registry: &mut KernelRegistry,
    device: &DeviceRef,
    y: &MlxBuffer,
    dy: &MlxBuffer,
    dx: &MlxBuffer,
    params: &MlxBuffer,
) -> Result<()>