| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | nadam.py | 127 local_step = math_ops.cast(self.iterations + 1, var_dtype)
130 1. - 0.5 * (math_ops.pow(decay_base, self._initial_decay * local_step)))
133 (math_ops.pow(decay_base, self._initial_decay * (local_step + 1))))
147 local_step = math_ops.cast(self.iterations + 1, var_dtype)
155 v_t_prime = v_t / (1. - math_ops.pow(beta_2_t, local_step))
167 local_step = math_ops.cast(self.iterations + 1, var_dtype)
190 v_t_prime = v_t_slice / (1. - math_ops.pow(beta_2_t, local_step))
|
| D | adam.py | 172 local_step = math_ops.cast(self.iterations + 1, var_dtype)
173 beta_1_power = math_ops.pow(beta_1_t, local_step)
174 beta_2_power = math_ops.pow(beta_2_t, local_step)
209 local_step = math_ops.cast(self.iterations + 1, var_dtype)
210 beta_1_power = math_ops.pow(beta_1_t, local_step)
211 beta_2_power = math_ops.pow(beta_2_t, local_step)
|
| D | adamax.py | 122 local_step = math_ops.cast(self.iterations + 1, var_dtype)
123 beta_1_power = math_ops.pow(beta_1_t, local_step)
142 local_step = math_ops.cast(self.iterations + 1, var_dtype)
143 beta_1_power = math_ops.pow(beta_1_t, local_step)
|
| D | nadam_test.py | 35 local_step = math_ops.cast(opt.iterations + 1, dtype)
37 beta_1_power = math_ops.pow(beta_1_t, local_step)
39 beta_2_power = math_ops.pow(beta_2_t, local_step)
|
| D | optimizer_v2.py | 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)
610 lr_t = lr_t / (1. + decay_t * local_step)
|
| D | adam_test.py | 103 local_step = math_ops.cast(opt.iterations + 1, dtype)
105 beta_1_power = math_ops.pow(beta_1_t, local_step)
107 beta_2_power = math_ops.pow(beta_2_t, local_step)
|
| D | adamax_test.py | 73 local_step = math_ops.cast(opt.iterations + 1, dtype)
75 beta_1_power = math_ops.pow(beta_1_t, local_step)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | conditional_accumulator_test.py | 253 accum_op = q.apply_grad((10.0,), local_step=1)
269 accum_ops = [q.apply_grad((x,), local_step=0) for x in elems]
278 accum_ops = [q.apply_grad((x,), local_step=1) for x in elems]
297 accum_ops = [q.apply_grad((x,), local_step=0) for x in elems]
306 accum_ops = [q.apply_grad((x,), local_step=1) for x in elems]
348 accum_ops = [q.apply_grad((x,), local_step=0) for x in elems]
359 accum_ops = [q.apply_grad((x,), local_step=1) for x in elems]
379 accum_ops = [q.apply_grad((x,), local_step=0) for x in elems]
390 accum_ops = [q.apply_grad((x,), local_step=1) for x in elems]
423 accum_ops = [q.apply_grad((0.0 + x,), local_step=x) for x in local_steps]
[all …]
|
| D | sparse_conditional_accumulator_test.py | 239 accum_op = q.apply_indexed_slices_grad(grad_indexed_slices, local_step=0)
244 local_step=0)
256 accum_op = q.apply_indexed_slices_grad(grad_indexed_slices, local_step=1)
261 local_step=1)
279 accum_ops.append(q.apply_indexed_slices_grad(x, local_step=0))
314 accum_ops.append(q.apply_indexed_slices_grad(x, local_step=0))
346 accum_ops.append(q.apply_indexed_slices_grad(v, local_step=e - 1))
388 accum_ops.append(q.apply_indexed_slices_grad(x, local_step=0))
619 local_step=1).run()
628 local_step=1).run()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_AccumulatorApplyGradient.pbtxt | 10 name: "local_step"
12 The local_step value at which the gradient was computed.
30 Does not add if local_step is lesser than the accumulator's global_step.
|
| D | api_def_SparseAccumulatorApplyGradient.pbtxt | 10 name: "local_step"
12 The local_step value at which the sparse gradient was computed.
52 Does not add if local_step is smaller than the accumulator's
|
| D | api_def_SparseConditionalAccumulator.pbtxt | 37 The accumulator accepts gradients marked with local_step greater or
|
| D | api_def_ConditionalAccumulator.pbtxt | 37 The accumulator accepts gradients marked with local_step greater or
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | typed_conditional_accumulator_base.h | 52 void TryApplyGrad(int64 local_step, OpKernelContext* ctx) override { in TryApplyGrad() argument
55 if (local_step >= current_global_step_) { in TryApplyGrad()
|
| D | conditional_accumulator_base.h | 59 virtual void TryApplyGrad(int64 local_step, OpKernelContext* ctx) = 0;
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | data_flow_ops.py | 1275 def apply_grad(self, grad, local_step=0, name=None): argument
1294 local_step = math_ops.cast(ops.convert_to_tensor(local_step), _dtypes.int64)
1296 self._accumulator_ref, local_step=local_step, gradient=grad, name=name)
1365 def apply_indexed_slices_grad(self, grad, local_step=0, name=None): argument
1386 local_step=local_step,
1393 local_step=0, argument
1426 local_step = math_ops.cast(ops.convert_to_tensor(local_step), _dtypes.int64)
1429 local_step=local_step,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.sparse.-sparse-conditional-accumulator.pbtxt | 24 …argspec: "args=[\'self\', \'grad_indices\', \'grad_values\', \'grad_shape\', \'local_step\', \'nam…
28 …argspec: "args=[\'self\', \'grad\', \'local_step\', \'name\'], varargs=None, keywords=None, defaul…
|
| D | tensorflow.-sparse-conditional-accumulator.pbtxt | 24 …argspec: "args=[\'self\', \'grad_indices\', \'grad_values\', \'grad_shape\', \'local_step\', \'nam…
28 …argspec: "args=[\'self\', \'grad\', \'local_step\', \'name\'], varargs=None, keywords=None, defaul…
|
| D | tensorflow.-conditional-accumulator.pbtxt | 24 …argspec: "args=[\'self\', \'grad\', \'local_step\', \'name\'], varargs=None, keywords=None, defaul…
|
| D | tensorflow.raw_ops.pbtxt | 17 …argspec: "args=[\'handle\', \'local_step\', \'gradient\', \'name\'], varargs=None, keywords=None, …
3409 …argspec: "args=[\'handle\', \'local_step\', \'gradient_indices\', \'gradient_values\', \'gradient_…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.sparse.-sparse-conditional-accumulator.pbtxt | 24 …argspec: "args=[\'self\', \'grad_indices\', \'grad_values\', \'grad_shape\', \'local_step\', \'nam…
28 …argspec: "args=[\'self\', \'grad\', \'local_step\', \'name\'], varargs=None, keywords=None, defaul…
|
| /external/tensorflow/tensorflow/python/training/ |
| D | moving_averages.py | 228 local_step = variable_scope.get_variable(
239 update_local_step = local_step.assign_add(1)
247 1.0 - decay, local_step.read_value())))
|
| D | sync_replicas_optimizer.py | 291 grad, local_step=self._local_step))
300 grad, local_step=self._local_step))
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | model_average_optimizer.py | 240 var_accum.apply_grad(var._ref(), local_step=global_step))
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | layers_test.py | 1939 local_step = variables.get_variables_by_name('local_step')[0]
1943 self.assertEqual(local_step.op.name,
2033 local_step = variables.get_variables_by_name('local_step')[0]
2036 self.assertAllClose(local_step.eval(), 0)
2041 self.assertAllClose(local_step.eval(), i)
|