Searched defs:beta1 (Results 1 - 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dtraining_ops_gpu.cu.cc109 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 Dtraining_ops_test.cc175 auto beta1 = Scalar(g, 0.9); local
181 {var, m, v, beta1_power, beta2_power, lr, beta1, beta2, epsilon, grad});
H A Dtraining_ops.cc290 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 Dtraining_ops.cc194 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 Dsac_reshapeBBEnv.cpp365 /*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 Dcvsurf.cpp423 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