fast_primitives.h File Reference

#include <optional>
#include "mlx/primitives.h"

Go to the source code of this file.

Classes
class	mlx::core::fast::Custom
class	mlx::core::fast::RMSNorm
class	mlx::core::fast::RMSNormVJP
class	mlx::core::fast::LayerNorm
class	mlx::core::fast::LayerNormVJP
class	mlx::core::fast::RoPE
class	mlx::core::fast::ScaledDotProductAttention
class	mlx::core::fast::AffineQuantize
struct	mlx::core::fast::CustomKernelShapeInfo
class	mlx::core::fast::CustomKernel

Namespaces
namespace	mlx
namespace	mlx::core
namespace	mlx::core::fast