diff --git a/tests/tt_eager/python_api_testing/sweep_tests/pytorch_ops.py b/tests/tt_eager/python_api_testing/sweep_tests/pytorch_ops.py index 9aaa1f58b30..b2be7ec52c4 100644 --- a/tests/tt_eager/python_api_testing/sweep_tests/pytorch_ops.py +++ b/tests/tt_eager/python_api_testing/sweep_tests/pytorch_ops.py @@ -1865,6 +1865,7 @@ def max_pool2d(x, *args, **kwargs): def repeat_2(x, *args, shape, **kwargs): return x.repeat(*shape) + def power_2(x, y, *args, exponent=None, **kwargs): if exponent is None: result = torch.pow(x, y) @@ -1872,6 +1873,7 @@ def power_2(x, y, *args, exponent=None, **kwargs): result = x**exponent return result + def subtract_and_apply_activation(x, y, *args, **kwargs): activation = kwargs.pop("activation") output = torch.sub(x, y)