Searched refs:alpha_t (Results 1 – 10 of 10) sorted by relevance
137 alpha_t = math_ops.cast(self._alpha_t, var.dtype.base_dtype)153 multiplier_values = alpha_t + sign_decayed * sign_gm.values
41 alpha_t = alpha * np.sqrt(1 - beta2**t) / (1 - beta1**t)48 param_t = param - alpha_t * m_bar / (np.sqrt(v_t) + epsilon)
47 alpha_t = alpha * np.sqrt(1 - beta2**t) / (1 - beta1**t)52 param_t = param - alpha_t * m_t / (np.sqrt(v_t) + epsilon)
43 alpha_t = alpha * np.sqrt(1 - beta2**t) / (1 - beta1**t)48 param_t = param - alpha_t * m_t / (np.sqrt(v_t) + epsilon)
276 const Tensor& alpha_t = ctx->input(1); in Compute() local297 samples_shape.AppendShape(alpha_t.shape()); in Compute()319 const auto alpha_flat = alpha_t.flat<T>().data(); in Compute()320 const int64 num_alphas = alpha_t.NumElements(); in Compute()
46 alpha_t = alpha * np.sqrt(1 - beta2**t) / (1 - beta1**t)51 param_t = param - alpha_t * m_t / (np.sqrt(v_t) + epsilon)
301 alpha_t = alpha * np.sqrt(1 - beta2**t) / (1 - beta1**t)306 param_t = param - alpha_t * m_t / (np.sqrt(v_t) + epsilon)