| /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_ResourceAccumulatorApplyGradient.pbtxt | 11 name: "local_step"
13 The local_step value at which the gradient was computed.
31 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
|
| D | api_def_ResourceConditionalAccumulator.pbtxt | 38 The accumulator accepts gradients marked with local_step greater or
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | conditional_accumulator_test.py | 218 accum_op = q.apply_grad((10.0,), local_step=1)
234 accum_ops = [q.apply_grad((x,), local_step=0) for x in elems]
243 accum_ops = [q.apply_grad((x,), local_step=1) for x in elems]
262 accum_ops = [q.apply_grad((x,), local_step=0) for x in elems]
271 accum_ops = [q.apply_grad((x,), local_step=1) for x in elems]
313 accum_ops = [q.apply_grad((x,), local_step=0) for x in elems]
324 accum_ops = [q.apply_grad((x,), local_step=1) for x in elems]
344 accum_ops = [q.apply_grad((x,), local_step=0) for x in elems]
355 accum_ops = [q.apply_grad((x,), local_step=1) for x in elems]
388 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)
282 accum_ops.append(q.apply_indexed_slices_grad(x, local_step=0))
320 accum_ops.append(q.apply_indexed_slices_grad(x, local_step=0))
355 accum_ops.append(q.apply_indexed_slices_grad(v, local_step=e - 1))
400 accum_ops.append(q.apply_indexed_slices_grad(x, local_step=0))
634 local_step=1).run()
643 local_step=1).run()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | AccumulatorApplyGradient.pbtxt | 9 name: "local_step"
47 name: "local_step"
87 name: "local_step"
128 name: "local_step"
|
| D | SparseAccumulatorApplyGradient.pbtxt | 9 name: "local_step"
59 name: "local_step"
111 name: "local_step"
164 name: "local_step"
|
| D | ResourceAccumulatorApplyGradient.pbtxt | 8 name: "local_step"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | AccumulatorApplyGradient.pbtxt | 9 name: "local_step"
47 name: "local_step"
87 name: "local_step"
128 name: "local_step"
|
| D | SparseAccumulatorApplyGradient.pbtxt | 9 name: "local_step"
59 name: "local_step"
111 name: "local_step"
164 name: "local_step"
|
| D | ResourceAccumulatorApplyGradient.pbtxt | 8 name: "local_step"
|
| /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()
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | adam.py | 137 local_step = math_ops.cast(self.iterations + 1, var_dtype)
140 beta_1_power = math_ops.pow(beta_1_t, local_step)
141 beta_2_power = math_ops.pow(beta_2_t, local_step)
389 local_step = math_ops.cast(self.iterations + 1, var_dtype)
392 beta_1_power = math_ops.pow(beta_1_t, local_step)
393 beta_2_power = math_ops.pow(beta_2_t, local_step)
|
| D | nadam_test.py | 33 local_step = math_ops.cast(opt.iterations + 1, dtype)
35 beta_1_power = math_ops.pow(beta_1_t, local_step)
37 beta_2_power = math_ops.pow(beta_2_t, local_step)
|
| D | nadam.py | 113 local_step = math_ops.cast(self.iterations + 1, var_dtype)
119 math_ops.pow(decay_base, self._initial_decay * local_step)))
142 v_t_prime_denominator=1. - math_ops.pow(beta_2_t, local_step),
|
| D | adamax.py | 116 local_step = math_ops.cast(self.iterations + 1, var_dtype)
119 beta_1_power = math_ops.pow(beta_1_t, local_step)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| 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.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.-conditional-accumulator.pbtxt | 24 …argspec: "args=[\'self\', \'grad\', \'local_step\', \'name\'], varargs=None, keywords=None, defaul…
|
| /external/tensorflow/tensorflow/python/training/ |
| D | moving_averages.py | 263 local_step = variable_scope.get_variable(
270 def update_fn(v, value, biased_var, local_step): argument
273 update_local_step = local_step.assign_add(1)
281 strategy, unbiased_var, update_fn, args=(value, biased_var, local_step))
|
| D | sync_replicas_optimizer.py | 289 grad, local_step=self._local_step))
298 grad, local_step=self._local_step))
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | data_flow_ops.py | 1358 def apply_grad(self, grad, local_step=0, name=None): argument
1377 local_step = math_ops.cast(ops.convert_to_tensor(local_step), _dtypes.int64)
1380 self._accumulator_ref, local_step=local_step, gradient=grad, name=name)
1447 def apply_indexed_slices_grad(self, grad, local_step=0, name=None): argument
1468 local_step=local_step,
1475 local_step=0, argument
1508 local_step = math_ops.cast(ops.convert_to_tensor(local_step), _dtypes.int64)
1511 local_step=local_step,
|