diff --git a/tensorflow/core/kernels/relu_op.h b/tensorflow/core/kernels/relu_op.h index 26b417950b7..4df9ebfad53 100644 --- a/tensorflow/core/kernels/relu_op.h +++ b/tensorflow/core/kernels/relu_op.h @@ -150,7 +150,7 @@ class LeakyReluGradOp Tensor* output) { if (!ReluHelpers::ValidateSameSize(context, g, a)) return; functor::LeakyReluGrad functor; - functor(context->eigen_device(), g.flat(), a.flat(), alpha, + functor(context->eigen_device(), g.flat(), a.flat(), alpha_, output->flat()); }