Commit a79e7cf6 authored by Davis King's avatar Davis King

fixed compile time bugs

parent cb2f9de6
......@@ -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;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment