fast_primitives.h File Reference

#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

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