#18843: Update BN sweep test

tenstorrent · Mar 10, 2025 · 43d263e · 43d263e
1 parent 53a2df1
commit 43d263e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/sweep_framework/sweeps/normalization/batch_norm/batch_norm.py b/tests/sweep_framework/sweeps/normalization/batch_norm/batch_norm.py
@@ -24,7 +24,7 @@
 
 parameters = {
     "BN_Testing": {
-        "input_shape": gen_shapes([1, 1, 32, 32], [6, 12, 256, 256], [1, 1, 32, 32], 16),
+        "input_shape": gen_shapes([1, 1, 32, 32], [6, 12, 256, 256], [2, 4, 64, 64], 4),
         "input_dtype": [ttnn.bfloat16, ttnn.float32],
         "input_layout": [ttnn.TILE_LAYOUT],
         "input_memory_config": [ttnn.DRAM_MEMORY_CONFIG, ttnn.L1_MEMORY_CONFIG],