use LARGE_X to define the input shape

apache · Oct 8, 2019 · f0c3497 · f0c3497
1 parent dfe7ac2
commit f0c3497
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/nightly/test_large_array.py b/tests/nightly/test_large_array.py
@@ -813,7 +813,7 @@ def get_ref_mean_var(data, running_mean, running_var, eps, use_global_status=Tru
         nd.waitall()
         return mean, stdvar
 
-    shape = (MEDIUM_X, MEDIUM_X, SMALL_Y, SMALL_Y)
+    shape = (3, 3, LARGE_X, SMALL_Y)
     axis = 1  # default
     eps = 1e-3