| /external/tensorflow/tensorflow/core/kernels/ |
| training_ops_test.cc | 173 auto beta2_power = Scalar(g, 0.99); local
181 {var, m, v, beta1_power, beta2_power, lr, beta1, beta2, epsilon, grad});
|
| training_ops.cc | 288 typename TTypes<T>::ConstScalar beta2_power,
294 const T alpha = lr() * Eigen::numext::sqrt(T(1) - beta2_power()) /
317 T beta1_power, T beta2_power, T lr, T beta1, T beta2,
320 lr * Eigen::numext::sqrt(T(1) - beta2_power) / (T(1) - beta1_power);
2515 const Tensor& beta2_power = ctx->input(4); variable
2612 T beta2_power = 0; variable
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| training_ops.cc | 188 errors::InvalidArgument("beta2_power is not a scalar: ",
217 xla::ComputationDataHandle beta2_power = ctx->Input(4); variable
235 b->Div(b->Mul(lr, b->Pow(b->Sub(one, beta2_power), half)),
|
