Home
last modified time | relevance | path

Searched refs:var_dtype (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Dnadam.py107 var_dtype = var_list[0].dtype.base_dtype
112 dtype=var_dtype,
125 var_dtype = var_list[0].dtype.base_dtype
126 beta_1_t = self._get_hyper('beta_1', var_dtype)
127 local_step = math_ops.cast(self.iterations + 1, var_dtype)
128 decay_base = math_ops.cast(0.96, var_dtype)
140 var_dtype = var.dtype.base_dtype
141 lr_t = self._get_hyper('learning_rate', var_dtype)
142 epsilon_t = self._get_hyper('epsilon', var_dtype)
145 beta_1_t = self._get_hyper('beta_1', var_dtype)
[all …]
Dftrl.py147 var_dtype = var.dtype.base_dtype
148 lr_t = self._decayed_lr(var_dtype)
149 learning_rate_power = self._get_hyper('learning_rate_power', var_dtype)
151 var_dtype)
153 var_dtype)
176 math_ops.cast(self._l2_shrinkage_regularization_strength, var_dtype),
181 var_dtype = var.dtype.base_dtype
182 lr_t = self._decayed_lr(var_dtype)
183 learning_rate_power = self._get_hyper('learning_rate_power', var_dtype)
185 var_dtype)
[all …]
Dadamax.py116 var_dtype = var.dtype.base_dtype
117 lr_t = self._decayed_lr(var_dtype)
120 beta_1_t = self._get_hyper('beta_1', var_dtype)
121 beta_2_t = self._get_hyper('beta_2', var_dtype)
122 local_step = math_ops.cast(self.iterations + 1, var_dtype)
132 self._get_hyper('epsilon', var_dtype),
137 var_dtype = var.dtype.base_dtype
138 lr_t = self._decayed_lr(var_dtype)
140 beta_1_t = self._get_hyper('beta_1', var_dtype)
141 beta_2_t = self._get_hyper('beta_2', var_dtype)
[all …]
Dadam.py165 var_dtype = var.dtype.base_dtype
166 lr_t = self._decayed_lr(var_dtype)
169 beta_1_t = self._get_hyper('beta_1', var_dtype)
170 beta_2_t = self._get_hyper('beta_2', var_dtype)
171 epsilon = self._get_hyper('epsilon', var_dtype)
172 local_step = math_ops.cast(self.iterations + 1, var_dtype)
205 var_dtype = var.dtype.base_dtype
206 lr_t = self._decayed_lr(var_dtype)
207 beta_1_t = self._get_hyper('beta_1', var_dtype)
208 beta_2_t = self._get_hyper('beta_2', var_dtype)
[all …]
Dadadelta.py120 var_dtype = var.dtype.base_dtype
121 lr_t = self._decayed_lr(var_dtype)
129 self._get_hyper('rho', var_dtype),
130 self._get_hyper('epsilon', var_dtype),
135 var_dtype = var.dtype.base_dtype
136 lr_t = self._decayed_lr(var_dtype)
144 self._get_hyper('rho', var_dtype),
145 self._get_hyper('epsilon', var_dtype),
Dgradient_descent.py102 var_dtype = var.dtype.base_dtype
103 lr_t = self._decayed_lr(var_dtype)
111 self._get_hyper("momentum", var_dtype),
123 var_dtype = var.dtype.base_dtype
124 lr_t = self._decayed_lr(var_dtype)
130 var_dtype = var.dtype.base_dtype
131 lr_t = self._decayed_lr(var_dtype)
139 self._get_hyper("momentum", var_dtype),
Drmsprop.py134 var_dtype = var.dtype.base_dtype
135 lr_t = self._decayed_lr(var_dtype)
137 rho = self._get_hyper("rho", var_dtype)
138 momentum = self._get_hyper("momentum", var_dtype)
139 epsilon = self._get_hyper("epsilon", var_dtype)
179 var_dtype = var.dtype.base_dtype
180 lr_t = self._decayed_lr(var_dtype)
182 rho = self._get_hyper("rho", var_dtype)
183 momentum = self._get_hyper("momentum", var_dtype)
184 epsilon = self._get_hyper("epsilon", var_dtype)
Dadagrad.py143 var_dtype = var.dtype.base_dtype
144 lr_t = self._decayed_lr(var_dtype)
145 epsilon = self._get_hyper('epsilon', var_dtype)
161 var_dtype = var.dtype.base_dtype
162 lr_t = self._decayed_lr(var_dtype)
163 epsilon = self._get_hyper('epsilon', var_dtype)
Doptimizer_v2.py601 def _decayed_lr(self, var_dtype): argument
603 lr_t = self._get_hyper("learning_rate", var_dtype)
605 local_step = math_ops.cast(self.iterations, var_dtype)
606 lr_t = math_ops.cast(lr_t(local_step), var_dtype)
608 local_step = math_ops.cast(self.iterations, var_dtype)
609 decay_t = self._get_hyper("decay", var_dtype)
/external/tensorflow/tensorflow/contrib/opt/python/training/
Dggt.py212 var_dtype = self._variables[0].dtype.base_dtype
218 beta1 = state.get_hyper("beta1", dtype=var_dtype)
234 eps = state.get_hyper("eps", dtype=var_dtype)
235 svd_eps = state.get_hyper("svd_eps", dtype=var_dtype)
236 sigma_eps = state.get_hyper("sigma_eps", dtype=var_dtype)
237 lr = state.get_hyper("lr", dtype=var_dtype)
241 ops.convert_to_tensor(math_ops.cast(window, dtype=var_dtype))))
251 damping = math_ops.cast(linalg_ops.eye(window), dtype=var_dtype) * svd_eps
260 math_ops.cast(1.0, dtype=var_dtype),
268 array_ops.zeros(flat_grad.get_shape(), dtype=var_dtype), -1)
[all …]