Fixed undeclared identifier 'alpha'

This commit is contained in:
Artem Mavrin 2020-02-06 23:54:18 -08:00
parent 198cba7bca
commit 108d1e9202

View File

@ -150,7 +150,7 @@ class LeakyReluGradOp
Tensor* output) {
if (!ReluHelpers::ValidateSameSize(context, g, a)) return;
functor::LeakyReluGrad<Device, T> functor;
functor(context->eigen_device<Device>(), g.flat<T>(), a.flat<T>(), alpha,
functor(context->eigen_device<Device>(), g.flat<T>(), a.flat<T>(), alpha_,
output->flat<T>());
}