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: