Commit d059f3c1 authored by Davis King's avatar Davis King

fixed compiler errors

parent b084076e
...@@ -125,7 +125,7 @@ namespace ...@@ -125,7 +125,7 @@ namespace
auto f = [&](float eps) { auto f = [&](float eps) {
const float old = src.host()[idx]; const float old = src.host()[idx];
src.host()[idx] += eps; src.host()[idx] += eps;
softmax(dest, src); tt::softmax(dest, src);
float result = dot(gradient_input, dest); float result = dot(gradient_input, dest);
src.host()[idx] = old; src.host()[idx] = old;
return result; return result;
...@@ -138,7 +138,7 @@ namespace ...@@ -138,7 +138,7 @@ namespace
src_grad.copy_size(src); src_grad.copy_size(src);
src_grad = 0; src_grad = 0;
softmax(dest, src); tt::softmax(dest, src);
softmax_gradient(src_grad, dest, gradient_input); softmax_gradient(src_grad, dest, gradient_input);
auto grad_error = compare_gradients(src_grad, grad_src); auto grad_error = compare_gradients(src_grad, grad_src);
......
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