Skip to content

Commit

Permalink
fix: fixed reciprocal_no_nan at tf frontend
Browse files Browse the repository at this point in the history
  • Loading branch information
samthakur587 committed Mar 13, 2024
1 parent 5194eab commit a60d5df
Showing 1 changed file with 4 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -2202,6 +2202,8 @@ def test_tensorflow_reciprocal_no_nan(
on_device,
):
input_dtype, x = dtype_and_x
if backend_fw == "paddle":
assume(not np.any(np.less_equal(x, 1e-08)))
helpers.test_frontend_function(
input_dtypes=input_dtype,
backend_to_test=backend_fw,
Expand Down Expand Up @@ -2372,6 +2374,8 @@ def test_tensorflow_reduce_max(
fn_tree="tensorflow.math.reduce_mean",
dtype_and_x=helpers.dtype_and_values(
available_dtypes=helpers.get_dtypes("float"),
min_value=-1e30,
max_value=1e30,
),
test_with_out=st.just(False),
)
Expand Down

0 comments on commit a60d5df

Please sign in to comment.