mlx_fast_layer_norm function

  1. @Native<Int Function(Pointer<mlx_array>, mlx_array, mlx_array, mlx_array, Float, mlx_stream)>(ffi.Pointer<mlx_array>, mlx_array, mlx_array, mlx_array, ffi.Float, mlx_stream)>()
int mlx_fast_layer_norm(
  1. Pointer<mlx_array> res,
  2. mlx_array x,
  3. mlx_array weight,
  4. mlx_array bias,
  5. double eps,
  6. mlx_stream s,
)

Implementation

@ffi.Native<
  ffi.Int Function(
    ffi.Pointer<mlx_array>,
    mlx_array,
    mlx_array,
    mlx_array,
    ffi.Float,
    mlx_stream,
  )
>()
external int mlx_fast_layer_norm(
  ffi.Pointer<mlx_array> res,
  mlx_array x,
  mlx_array weight,
  mlx_array bias,
  double eps,
  mlx_stream s,
);