Speeding up data dumping routine

llvm/projects/keras/frontend/weight_utils.py

@@ -26,7 +26,7 @@ def dumpLabels(file_name, Y_test):
     
 
     
-def dumpData(file_name, X_test):
+"""def dumpData(file_name, X_test):
 
     N = X_test.shape[0]
     C = X_test.shape[1]
@@ -46,7 +46,27 @@ def dumpData(file_name, X_test):
                     f.write(np.float32(val[0]))
 
     f.close()
+"""
+
 
+def dumpData(file_name, X_test):
+
+    N = X_test.shape[0]
+    C = X_test.shape[1]
+    H = X_test.shape[2]
+    W = X_test.shape[3]
+ 
+    print ("*DumpData")
+    print("-min_val = ", np.amin(X_test))
+    print("-max_val = ", np.amax(X_test))
+
+    f = open(file_name, "wb")
+
+    X_test.tofile(f)    
+
+    f.close()
+
+    
     
 def dumpConvWeights(file_name, weights, N, C, H, W):