From f0c34973d24ce0e59f1494a81909b2cd9bf93c72 Mon Sep 17 00:00:00 2001
From: wuxun-zhang <wuxun.zhang@intel.com>
Date: Tue, 8 Oct 2019 21:10:57 +0800
Subject: [PATCH] use LARGE_X to define the input shape

---
 tests/nightly/test_large_array.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/nightly/test_large_array.py b/tests/nightly/test_large_array.py
index 09e9864e650a..6b0c42d21efb 100644
--- 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