Commit d059f3c1 authored by Davis King's avatar Davis King

fixed compiler errors

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