TransformScale< T > Struct Template Reference

#include <steel_attention.h>

Public Member Functions
METAL_FUNC	TransformScale (T scale_)
METAL_FUNC T	apply (T x) const

Public Attributes
T	scale

Constructor & Destructor Documentation

TransformScale()

template<typename T>

METAL_FUNC TransformScale< T >::TransformScale ( T scale_ )

inline

Member Function Documentation

apply()

template<typename T>

METAL_FUNC T TransformScale< T >::apply ( T x ) const

inline

Member Data Documentation

scale

template<typename T>

T TransformScale< T >::scale

---

The documentation for this struct was generated from the following file:

• mlx/backend/metal/kernels/steel/attn/kernels/steel_attention.h