Searched defs:beta1 (Results 1 - 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
H A D | training_ops_gpu.cu.cc | 109 typename TTypes<T>::ConstScalar beta1, 118 m + (beta1.constant(one) - beta1).reshape(single).broadcast(bcast) * 130 (m * beta1.reshape(single).broadcast(bcast) + 131 (beta1.constant(one) - beta1).reshape(single).broadcast(bcast) * 104 operator ()(const GPUDevice& d, typename TTypes<T>::Flat var, typename TTypes<T>::Flat m, typename TTypes<T>::Flat v, typename TTypes<T>::ConstScalar beta1_power, typename TTypes<T>::ConstScalar beta2_power, typename TTypes<T>::ConstScalar lr, typename TTypes<T>::ConstScalar beta1, typename TTypes<T>::ConstScalar beta2, typename TTypes<T>::ConstScalar epsilon, typename TTypes<T>::ConstFlat grad, bool use_nesterov) argument
|
H A D | training_ops_test.cc | 175 auto beta1 = Scalar(g, 0.9); local 181 {var, m, v, beta1_power, beta2_power, lr, beta1, beta2, epsilon, grad});
|
H A D | training_ops.cc | 290 typename TTypes<T>::ConstScalar beta1, 296 // beta1 == μ 301 m.device(d) += (grad - m) * (T(1) - beta1()); 304 var.device(d) -= ((grad * (T(1) - beta1()) + beta1() * m) * alpha) / 317 T beta1_power, T beta2_power, T lr, T beta1, T beta2, 321 m.device(d) += (grad - m) * (T(1) - beta1); 2517 const Tensor& beta1 = ctx->input(6); variable 2530 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(beta1.shape()), 2531 errors::InvalidArgument("beta1 i 285 operator ()(const Device& d, typename TTypes<T>::Flat var, typename TTypes<T>::Flat m, typename TTypes<T>::Flat v, typename TTypes<T>::ConstScalar beta1_power, typename TTypes<T>::ConstScalar beta2_power, typename TTypes<T>::ConstScalar lr, typename TTypes<T>::ConstScalar beta1, typename TTypes<T>::ConstScalar beta2, typename TTypes<T>::ConstScalar epsilon, typename TTypes<T>::ConstFlat grad, bool use_nesterov) argument 315 operator ()(const SYCLDevice& d, typename TTypes<T>::Flat var, typename TTypes<T>::Flat m, typename TTypes<T>::Flat v, T beta1_power, T beta2_power, T lr, T beta1, T beta2, T epsilon, typename TTypes<T>::ConstFlat grad) argument 2614 T beta1 = 0; variable [all...] |
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | training_ops.cc | 194 errors::InvalidArgument("beta1 is not a scalar: ", 219 xla::ComputationDataHandle beta1 = ctx->Input(6); variable 224 // alpha <- learning_rate * sqrt(1 - beta2^t) / (1 - beta1^t) 225 // m_t <- beta1 * m_{t-1} + (1 - beta1) * g_t 237 m = b->Add(m, b->Mul(b->Sub(grad, m), b->Sub(one, beta1)));
|
/external/aac/libSACdec/src/ |
H A D | sac_reshapeBBEnv.cpp | 365 /*FIXP_DBL beta1 = (FL2FXCONST_DBL(1.0f) - beta) << SF_BETA1;*/ 366 FIXP_DBL beta1 = ((FIXP_DBL)MAXVAL_DBL - beta) << SF_BETA1; local 533 normNrg = ((fMultDiv2(beta1, env) >> scaleCur) +
|
/external/opencv/cv/src/ |
H A D | cvsurf.cpp | 423 float alpha1 = (alpha0 - beta0)*sqrt_2*0.5f, beta1 = (alpha0 + beta0)*sqrt_2*0.5f; local 430 a = alpha1, b = beta1; 432 a = -beta1, b = alpha1;
|
Completed in 211 milliseconds