Searched refs:scale_alpha (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | elu_op.cc | 71 const auto scale_alpha = XlaHelpers::FloatLiteral(b, input_type(0), in Compile() local
76 xla::Mul(scale_alpha, expm1))); in Compile()
90 const auto scale_alpha = XlaHelpers::FloatLiteral(b, input_type(0), in Compile() local
95 const auto exp_grad = xla::Mul(grad, xla::Add(activation, scale_alpha)); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | relu_op_functor.h | 169 const auto scale_alpha = static_cast<T>(1.7580993408473768599402175208123); in operator() local
174 .select(scale_alpha * (features.exp() - features.constant(one)), in operator()
191 const auto scale_alpha = static_cast<T>(1.7580993408473768599402175208123); in operator() local
194 .select(gradients * (activations + scale_alpha), gradients * scale); in operator()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | relu_op_test.py | 503 scale_alpha = 1.7580993408473768599402175208123
504 return np.where(np_features < 0, scale_alpha * (np.exp(np_features) - 1),
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad.py | 422 scale_alpha = 1.7580993408473768599402175208123
425 x < 0., gen_nn_ops.elu_grad(grad, op.outputs[0] + scale_alpha),
|