fixed compile time bugs

a79e7cf6 · Davis King · cb2f9de6 · a79e7cf6
Commit a79e7cf6 authored Nov 09, 2015 by Davis King
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

dlib/dnn/tensor_tools.cpp dlib/dnn/tensor_tools.cpp +2 -1

No files found.
--- a/dlib/dnn/tensor_tools.cpp
+++ b/dlib/dnn/tensor_tools.cpp
@@ -5,6 +5,7 @@

 #include "tensor_tools.h"
 #include "cpu_dlib.h"
+#include "cuda_dlib.h"

 namespace dlib { namespace tt
 {
@@ -55,7 +56,7 @@ namespace dlib { namespace tt
    {
        DLIB_CASSERT(data.size()%2 == 0,"");
 #ifdef DLIB_USE_CUDA
-        rnd.fill_gaussian(data);
+        rnd.fill_gaussian(data, mean, stddev);
 #else
        for (auto& x : data) 
            x = rnd.get_random_gaussian()*stddev + mean;