Skip to content

Commit 28ee6db

Browse files
committed
fix: ivy.batch_norm test
1 parent 573e714 commit 28ee6db

File tree

1 file changed

+2
-0
lines changed
  • ivy_tests/test_ivy/test_functional/test_experimental/test_nn

1 file changed

+2
-0
lines changed

ivy_tests/test_ivy/test_functional/test_experimental/test_nn/test_norms.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -172,6 +172,8 @@ def _instance_and_batch_norm_helper(draw, *, min_dims=1, test_function="instance
172172
training=st.booleans(),
173173
test_instance_method=st.just(False),
174174
container_flags=st.just([False]),
175+
test_gradients=st.just(False),
176+
ground_truth_backend="numpy",
175177
)
176178
def test_batch_norm(*, data, training, test_flags, backend_fw, fn_name, on_device):
177179
x_dtype, x, mean, variance, offset, scale, eps, momentum, data_format = data

0 commit comments

Comments
 (0)