diff --git a/ivy/functional/frontends/tensorflow/math.py b/ivy/functional/frontends/tensorflow/math.py index f1a9ab0509358..cc1036143e371 100644 --- a/ivy/functional/frontends/tensorflow/math.py +++ b/ivy/functional/frontends/tensorflow/math.py @@ -20,6 +20,11 @@ def add(x, y, name=None): return ivy.add(x, y) +@to_ivy_arrays_and_back +def exp(x, name=None): + return ivy.exp(x) + + @to_ivy_arrays_and_back def argmax(input, axis, output_type=None, name=None): output_type = to_ivy_dtype(output_type) diff --git a/ivy_tests/test_ivy/test_frontends/test_tensorflow/test_math.py b/ivy_tests/test_ivy/test_frontends/test_tensorflow/test_math.py index cdc4e3cb5d205..d51f9dc249745 100644 --- a/ivy_tests/test_ivy/test_frontends/test_tensorflow/test_math.py +++ b/ivy_tests/test_ivy/test_frontends/test_tensorflow/test_math.py @@ -93,6 +93,31 @@ def test_tensorflow_tan( on_device=on_device, x=x[0], ) + + +# exp +@handle_frontend_test( + fn_tree="tensorflow.math.exp", + dtype_and_x=helpers.dtype_and_values(available_dtypes=helpers.get_dtypes("float")), + test_with_out=st.just(False), +) +def test_tensorflow_exp( + *, + dtype_and_x, + frontend, + test_flags, + fn_tree, + on_device, +): + input_dtype, x = dtype_and_x + helpers.test_frontend_function( + input_dtypes=input_dtype, + frontend=frontend, + test_flags=test_flags, + fn_tree=fn_tree, + on_device=on_device, + x=x[0], + ) # multiply