ScaleOp< OutT, InT > Struct Template Reference

#include <gemv_masked.h>

Public Member Functions
METAL_FUNC OutT	apply (InT x) const
METAL_FUNC OutT	apply (InT x) const

Public Attributes
OutT	scale

Member Function Documentation

apply() [1/2]

template<typename OutT , typename InT = OutT>

METAL_FUNC OutT ScaleOp< OutT, InT >::apply ( InT x ) const

inline

apply() [2/2]

template<typename OutT , typename InT = OutT>

METAL_FUNC OutT ScaleOp< OutT, InT >::apply ( InT x ) const

inline

Member Data Documentation

scale

template<typename OutT , typename InT = OutT>

OutT ScaleOp< OutT, InT >::scale

---

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

• mlx/backend/metal/kernels/gemv_masked.h
• mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_masked.h