# SPDX-License-Identifier: MIT
from torch import Tensor
from ..jit.core import compile_ops

MD_NAME = "module_aiter_operator"


@compile_ops("module_aiter_operator")
def add(input: Tensor, other: Tensor) -> Tensor: ...


@compile_ops("module_aiter_operator")
def sub(input: Tensor, other: Tensor) -> Tensor: ...


@compile_ops("module_aiter_operator")
def mul(input: Tensor, other: Tensor) -> Tensor: ...


@compile_ops("module_aiter_operator")
def div(input: Tensor, other: Tensor) -> Tensor: ...


@compile_ops("module_aiter_operator")
def add_(input: Tensor, other: Tensor) -> Tensor: ...


@compile_ops("module_aiter_operator")
def sub_(input: Tensor, other: Tensor) -> Tensor: ...


@compile_ops("module_aiter_operator")
def mul_(input: Tensor, other: Tensor) -> Tensor: ...


@compile_ops("module_aiter_operator")
def div_(input: Tensor, other: Tensor) -> Tensor: ...


@compile_ops("module_aiter_unary")
def sigmoid(input: Tensor) -> Tensor: ...


@compile_ops("module_aiter_unary")
def tanh(input: Tensor) -> Tensor: ...
