Searched refs:epsilon_t (Results 1 – 12 of 12) sorted by relevance
/external/tensorflow/tensorflow/contrib/opt/python/training/ |
D | lazy_adam_gs_optimizer.py | 57 epsilon_t = math_ops.cast(self._epsilon_t, var.dtype.base_dtype) 77 denominator_slice = math_ops.sqrt(v_t_slice) + epsilon_t 90 epsilon_t = math_ops.cast(self._epsilon_t, var.dtype.base_dtype) 109 var_slice = lr * m_t_slice / (math_ops.sqrt(v_t_slice) + epsilon_t)
|
D | lazy_adam_optimizer.py | 57 epsilon_t = math_ops.cast(self._epsilon_t, var.dtype.base_dtype) 77 denominator_slice = math_ops.sqrt(v_t_slice) + epsilon_t 90 epsilon_t = math_ops.cast(self._epsilon_t, var.dtype.base_dtype) 109 var_slice = lr * m_t_slice / (math_ops.sqrt(v_t_slice) + epsilon_t)
|
D | nadam_optimizer.py | 78 epsilon_t = math_ops.cast(self._epsilon_t, var.dtype.base_dtype) 96 var_update = scatter_add(var, indices, -lr * m_bar / (v_sqrt + epsilon_t))
|
D | adamax.py | 144 epsilon_t = math_ops.cast(self._epsilon_t, var.dtype.base_dtype) 159 (v_t_slice + epsilon_t))
|
D | adam_gs_optimizer.py | 186 epsilon_t = math_ops.cast(self._epsilon_t, var.dtype.base_dtype) 202 var, lr * m_t / (v_sqrt + epsilon_t), use_locking=self._use_locking)
|
/external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
D | nadam.py | 142 epsilon_t = self._get_hyper('epsilon', var_dtype) 157 var_t = var - lr_t * m_t_bar / (math_ops.sqrt(v_t_prime) + epsilon_t) 163 epsilon_t = self._get_hyper('epsilon', var_dtype) 194 var, indices, -lr_t * m_t_bar / (v_prime_sqrt + epsilon_t))
|
D | adam.py | 212 epsilon_t = self._get_hyper('epsilon', var_dtype) 232 var, lr * m_t / (v_sqrt + epsilon_t), use_locking=self._use_locking) 243 lr * m_t / (v_hat_sqrt + epsilon_t),
|
D | adamax.py | 144 epsilon_t = self._get_hyper('epsilon', var_dtype) 161 m_t_slice / (v_t_slice + epsilon_t))
|
/external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
D | quantile_ops.cc | 108 const Tensor* epsilon_t; in Compute() local 109 OP_REQUIRES_OK(context, context->input(kEpsilonName, &epsilon_t)); in Compute() 110 float epsilon = epsilon_t->scalar<float>()(); in Compute() 163 const Tensor* epsilon_t; in Compute() local 164 OP_REQUIRES_OK(context, context->input(kEpsilonName, &epsilon_t)); in Compute() 165 float epsilon = epsilon_t->scalar<float>()(); in Compute()
|
/external/tensorflow/tensorflow/contrib/optimizer_v2/ |
D | adam.py | 156 epsilon_t = state.get_hyper("epsilon", var.dtype.base_dtype) 172 var, lr * m_t / (v_sqrt + epsilon_t), use_locking=self._use_locking)
|
/external/tensorflow/tensorflow/python/training/ |
D | adam.py | 186 epsilon_t = math_ops.cast(self._epsilon_t, var.dtype.base_dtype) 202 var, lr * m_t / (v_sqrt + epsilon_t), use_locking=self._use_locking)
|
D | training_ops_test.py | 287 epsilon_t = constant_op.constant(epsilon, self._toType(var.dtype), []) 295 beta2_t, epsilon_t, grad)
|