MLX
Loading...
Searching...
No Matches
Public Member Functions
|
Public Attributes
|
List of all members
ScaleOp< OutT, InT > Struct Template Reference
#include <
steel_gemm_masked.h
>
Public Member Functions
METAL_FUNC OutT
apply
(InT x) const
Public Attributes
OutT
scale
Member Function Documentation
◆
apply()
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 file:
mlx/backend/metal/kernels/steel/gemm/kernels/
steel_gemm_masked.h
Generated by
1.10.0