File size: 220 Bytes
be5e628
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from . import layers

from ._ops import ops


def apply_rms_norm(input, weight, eps):
    return ops.apply_rms_norm(
            input,
            weight,
            eps,
    )

__all__ = ["layers", "apply_rms_norm"]