Packages

emily

0.1.2
1.0.0 0.7.2 0.7.1 0.6.1 0.5.1 0.4.0 0.3.5 0.3.3 0.3.2 0.1.2

Elixir bindings and Nx backend for Apple MLX

HexDocs

Current section

Files

Jump to
Readme
10 Versions
6 Dependencies
Dependants
Files
Activity
emily vendor mlx mlx backend gpu scan.h
Raw

vendor/mlx/mlx/backend/gpu/scan.h 

#pragma once



#include "mlx/array.h"

#include "mlx/primitives.h"



namespace mlx::core {



void scan_gpu_inplace(

    array in,

    array& out,

    Scan::ReduceType reduce_type,

    int axis,

    bool reverse,

    bool inclusive,

    const Stream& s);



} // namespace mlx::core