File size: 189 Bytes
bb1e912 |
1 2 3 4 5 6 7 8 9 10 11 12 |
import torch
from ._ops import ops
from .silu_and_mul import _silu_and_mul
def silu_and_mul(x: torch.Tensor) -> torch.Tensor:
return ops.silu_and_mul(x)
__all__ = ["silu_and_mul"]
|