fixed compile time bugs

This commit is contained in:
Davis King 2015-11-09 20:25:35 -05:00
parent cb2f9de634
commit a79e7cf645
1 changed files with 2 additions and 1 deletions

View File

@ -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;