HomeSort by relevance Sort by last modified time
    Searched refs:scale_alpha (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
elu_op.cc 71 const auto scale_alpha = XlaHelpers::FloatLiteral(b, input_type(0), variable
76 xla::Mul(scale_alpha, expm1)));
90 const auto scale_alpha = XlaHelpers::FloatLiteral(b, input_type(0), variable
95 const auto exp_grad = xla::Mul(grad, xla::Add(activation, scale_alpha));
  /external/tensorflow/tensorflow/core/kernels/
relu_op_functor.h 169 const auto scale_alpha = static_cast<T>(1.7580993408473768599402175208123); local
174 .select(scale_alpha * (features.exp() - features.constant(one)),
191 const auto scale_alpha = static_cast<T>(1.7580993408473768599402175208123); local
194 .select(gradients * (activations + scale_alpha), gradients * scale);
  /external/tensorflow/tensorflow/python/kernel_tests/
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/
nn_grad.py 422 scale_alpha = 1.7580993408473768599402175208123
425 x < 0., gen_nn_ops.elu_grad(grad, op.outputs[0] + scale_alpha),
    [all...]

Completed in 574 milliseconds