/external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
D | legacy_learning_rate_decay.py | 33 global_step, argument 101 decayed_lr = decayed_lr(global_step) 103 decayed_lr = functools.partial(decayed_lr, global_step) 187 global_step, argument 280 decayed_lr = decayed_lr(global_step) 282 decayed_lr = functools.partial(decayed_lr, global_step) 288 global_step, argument 368 decayed_lr = decayed_lr(global_step) 370 decayed_lr = functools.partial(decayed_lr, global_step) 376 global_step, argument [all …]
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | adagrad_da_test.py | 37 global_step = resource_variable_ops.ResourceVariable( 45 global_step, 50 zip([grads0, grads1], [var0, var1]), global_step=global_step) 73 global_step = resource_variable_ops.ResourceVariable( 82 global_step, 87 zip([grads0, grads1], [var0, var1]), global_step=global_step) 104 global_step = resource_variable_ops.ResourceVariable( 113 global_step, 118 zip([grads0, grads1], [var0, var1]), global_step=global_step) 135 global_step = resource_variable_ops.ResourceVariable( [all …]
|
/external/tensorflow/tensorflow/python/training/ |
D | sync_replicas_optimizer_test.py | 44 global_step = variables.VariableV1( 69 global_step=global_step) 117 global_step = graphs[1].get_tensor_by_name("global_step:0") 120 self.assertAllEqual(0, sessions[1].run(global_step)) 132 while sessions[1].run(global_step) != 1: 153 self.assertAllEqual(1, sessions[1].run(global_step)) 177 self.assertAllEqual(2, sessions[1].run(global_step)) 200 global_step = graphs[1].get_tensor_by_name("global_step:0") 203 self.assertAllEqual(0, sessions[1].run(global_step)) 215 while sessions[1].run(global_step) != 1: [all …]
|
D | adagrad_da_test.py | 40 global_step = variables.Variable(0, dtype=dtypes.int64) 51 global_step, 56 zip([grads0, grads1], [var0, var1]), global_step=global_step) 89 global_step = resource_variable_ops.ResourceVariable( 95 1.0, global_step).minimize(loss) 109 global_step = variables.Variable(0, dtype=dtypes.int64) 117 global_step, 122 zip([grads0, grads1], [var0, var1]), global_step=global_step) 141 global_step = variables.Variable(0, dtype=dtypes.int64) 149 global_step, [all …]
|
D | supervisor.py | 220 global_step=USE_DEFAULT, argument 320 self._init_global_step(global_step=global_step) 486 def _init_global_step(self, global_step=USE_DEFAULT): argument 493 if global_step is Supervisor.USE_DEFAULT: 494 global_step = self._get_first_op_from_collection( 496 if global_step is None: 497 global_step = self._default_global_step_tensor() 498 if global_step is not None: 499 ops.add_to_collection(ops.GraphKeys.GLOBAL_STEP, global_step) 500 self._global_step = global_step [all …]
|
D | basic_session_run_hooks.py | 368 def _update_steps_per_run_variable(self, global_step, session): argument 369 steps = min(self._last_step - global_step, 374 global_step = session.run(self._global_step_tensor) 376 self._last_step = global_step + self._num_steps 377 self._update_steps_per_run_variable(global_step, session) 382 global_step = run_context.session.run(self._global_step_tensor) 383 if global_step >= self._last_step: 386 self._update_steps_per_run_variable(global_step, run_context.session) 426 global_step = session.run(self._global_step_tensor) 427 self._last_step = global_step + self._num_steps [all …]
|
D | adagrad_da.py | 50 global_step, argument 87 self._global_step = global_step 115 global_step = array_ops.identity(self._global_step_on_worker) 124 global_step, 131 global_step = array_ops.identity(self._global_step_on_worker) 140 global_step, 147 global_step = array_ops.identity(self._global_step_on_worker) 157 global_step, 164 global_step = array_ops.identity(self._global_step_on_worker) 174 global_step,
|
D | sync_replicas_optimizer.py | 224 def apply_gradients(self, grads_and_vars, global_step=None, name=None): argument 250 if global_step is None: 253 self._global_step = global_step 267 dtype=global_step.dtype.base_dtype, 270 self.local_step_init_op = state_ops.assign(self._local_step, global_step) 307 with ops.device(global_step.device), ops.name_scope(""): 309 global_step) 312 with ops.device(global_step.device), ops.name_scope(""): 315 global_step.dtype.base_dtype, 321 with ops.device(global_step.device), ops.name_scope(""): [all …]
|
D | basic_session_run_hooks_test.py | 72 def before_save(self, session, global_step): argument 75 def after_save(self, session, global_step): argument 80 def end(self, session, global_step): argument 153 global_step = training_util.get_or_create_global_step() 158 sess.run(state_ops.assign(global_step, 5)) 162 sess.run(state_ops.assign(global_step, 9)) 165 sess.run(state_ops.assign(global_step, 10)) 168 sess.run(state_ops.assign(global_step, 11)) 177 global_step = training_util.get_or_create_global_step() 182 sess.run(state_ops.assign(global_step, 5)) [all …]
|
D | training_util_test.py | 31 def _assert_global_step(self, global_step, expected_dtype=dtypes.int64): argument 32 self.assertEqual('%s:0' % ops.GraphKeys.GLOBAL_STEP, global_step.name) 33 self.assertEqual(expected_dtype, global_step.dtype.base_dtype) 34 self.assertEqual([], global_step.get_shape().as_list()) 67 global_step = training_util.create_global_step() 68 self._assert_global_step(global_step)
|
D | optimizer_test.py | 54 global_step = resource_variable_ops.ResourceVariable( 63 opt_op = sgd_op.minimize(loss, global_step, [var0, var1]) 76 global_step = variables.Variable( 81 global_step, [var0, var1], 103 global_step = variables.Variable( 107 cost, global_step, [var0, var1], grad_loss=grad_loss) 245 global_step = variables.Variable( 248 opt_op = sgd_op.minimize(cost, global_step, [var0, var1]) 261 global_step = variables.Variable( 264 opt_op = sgd_op.minimize(cost, global_step, [var0, var1])
|
D | optimizer.py | 363 def minimize(self, loss, global_step=None, var_list=None, argument 420 return self.apply_gradients(grads_and_vars, global_step=global_step, 539 def apply_gradients(self, grads_and_vars, global_step=None, name=None): argument 577 self._distributed_apply, args=(grads_and_vars, global_step, name)) 625 if global_step is None: 629 with ops.colocate_with(global_step): 631 global_step, resource_variable_ops.BaseResourceVariable): 635 global_step.handle, 636 ops.convert_to_tensor(1, dtype=global_step.dtype), 639 apply_updates = state_ops.assign_add(global_step, 1, name=name) [all …]
|
/external/tensorflow/tensorflow/python/summary/writer/ |
D | fake_summary_writer.py | 112 def add_graph(self, graph, global_step=None, graph_def=None): argument 114 if (global_step is not None) and (global_step < 0): 115 raise ValueError('Invalid global_step %s.' % global_step) 120 def add_meta_graph(self, meta_graph_def, global_step=None): argument 122 if (global_step is not None) and (global_step < 0): 123 raise ValueError('Invalid global_step %s.' % global_step) 127 def add_session_log(self, session_log, global_step=None): argument 131 def add_run_metadata(self, run_metadata, tag, global_step=None): argument 132 if (global_step is not None) and (global_step < 0): 133 raise ValueError('Invalid global_step %s.' % global_step)
|
D | writer.py | 101 def add_summary(self, summary, global_step=None): argument 142 self._add_event(event, global_step) 144 def add_session_log(self, session_log, global_step=None): argument 156 self._add_event(event, global_step) 158 def _add_graph_def(self, graph_def, global_step=None): argument 161 self._add_event(event, global_step) 163 def add_graph(self, graph, global_step=None, graph_def=None): argument 214 self._add_graph_def(true_graph_def, global_step) 229 def add_meta_graph(self, meta_graph_def, global_step=None): argument 249 self._add_event(event, global_step) [all …]
|
/external/tensorflow/tensorflow/python/debug/lib/ |
D | debug_utils.py | 32 global_step=-1): argument 62 run_options.debug_options.global_step = global_step 90 global_step=-1, argument 181 global_step=global_step) 197 global_step=global_step) 210 global_step=-1, argument 290 global_step=global_step)
|
/external/tensorflow/tensorflow/core/summary/ |
D | summary_file_writer.cc | 70 Status WriteTensor(int64 global_step, Tensor t, const string& tag, in WriteTensor() argument 73 e->set_step(global_step); in WriteTensor() 93 Status WriteScalar(int64 global_step, Tensor t, const string& tag) override { in WriteScalar() argument 95 e->set_step(global_step); in WriteScalar() 102 Status WriteHistogram(int64 global_step, Tensor t, in WriteHistogram() argument 105 e->set_step(global_step); in WriteHistogram() 112 Status WriteImage(int64 global_step, Tensor t, const string& tag, in WriteImage() argument 115 e->set_step(global_step); in WriteImage() 122 Status WriteAudio(int64 global_step, Tensor t, const string& tag, in WriteAudio() argument 125 e->set_step(global_step); in WriteAudio() [all …]
|
/external/tensorflow/tensorflow/core/kernels/ |
D | summary_interface.h | 38 virtual Status WriteTensor(int64 global_step, Tensor t, const string& tag, 41 virtual Status WriteScalar(int64 global_step, Tensor t, 44 virtual Status WriteHistogram(int64 global_step, Tensor t, 47 virtual Status WriteImage(int64 global_step, Tensor t, const string& tag, 50 virtual Status WriteAudio(int64 global_step, Tensor t, const string& tag, 53 virtual Status WriteGraph(int64 global_step,
|
/external/tensorflow/tensorflow/python/keras/tests/ |
D | tracking_util_with_v1_optimizers_test.py | 89 global_step=optimizer_step) 92 global_step=optimizer_step) 95 model(input_value), global_step=optimizer_step) 98 global_step=optimizer_step) 275 global_step=root.optimizer_step) 289 global_step=root.optimizer_step) 320 global_step=root.optimizer_step) 357 global_step=training_util.get_or_create_global_step()) 361 global_step=root.global_step) 380 session.run(root.global_step)) [all …]
|
/external/tensorflow/tensorflow/python/tpu/ |
D | async_checkpoint.py | 107 global_step = session.run(self._global_step_tensor) 127 self._save(session, global_step) 128 self._timer.update_last_triggered_step(global_step) 135 global_step = run_context.session.run(self._global_step_tensor) 136 if self._timer.should_trigger_for_step(global_step): 137 self._timer.update_last_triggered_step(global_step) 138 logging.info("Triggering checkpoint. %s", global_step) 139 if self._save(run_context.session, global_step): 169 self._get_saver().save(session, self._save_path, global_step=step)
|
/external/tensorflow/tensorflow/python/training/experimental/ |
D | loss_scale_optimizer.py | 156 def apply_gradients(self, grads_and_vars, global_step=None, name=None): argument 182 return self._optimizer.apply_gradients(grads_and_vars, global_step, name) 189 self._distributed_apply, args=(grads_and_vars, global_step, name)) 194 global_step=None, argument 220 return self._apply_gradients(distribution, grads_and_vars, global_step, 228 def _apply_gradients(self, distribution, grads_and_vars, global_step, name): argument 232 args=(grads_and_vars, global_step, name))
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.summary.-file-writer.pbtxt | 16 …argspec: "args=[\'self\', \'graph\', \'global_step\', \'graph_def\'], varargs=None, keywords=None,… 20 …argspec: "args=[\'self\', \'meta_graph_def\', \'global_step\'], varargs=None, keywords=None, defau… 24 …argspec: "args=[\'self\', \'run_metadata\', \'tag\', \'global_step\'], varargs=None, keywords=None… 28 …argspec: "args=[\'self\', \'session_log\', \'global_step\'], varargs=None, keywords=None, defaults… 32 …argspec: "args=[\'self\', \'summary\', \'global_step\'], varargs=None, keywords=None, defaults=[\'…
|
D | tensorflow.train.-adagrad-d-a-optimizer.pbtxt | 21 …argspec: "args=[\'self\', \'learning_rate\', \'global_step\', \'initial_gradient_squared_accumulat… 25 …argspec: "args=[\'self\', \'grads_and_vars\', \'global_step\', \'name\'], varargs=None, keywords=N… 45 …argspec: "args=[\'self\', \'loss\', \'global_step\', \'var_list\', \'gate_gradients\', \'aggregati…
|
/external/tensorflow/tensorflow/python/data/experimental/ops/ |
D | iterator_ops.py | 279 global_step = session.run(self._checkpoint_saver_hook._global_step_tensor) 280 self._checkpoint_saver_hook._save(session, global_step) 281 self._checkpoint_saver_hook._timer.update_last_triggered_step(global_step) 311 global_step=None, argument 318 sess, save_path, global_step, latest_filename or self._latest_filename,
|
/external/tensorflow/tensorflow/core/api_def/base_api/ |
D | api_def_AccumulatorSetGlobalStep.pbtxt | 12 The new global_step value to set. 15 summary: "Updates the accumulator with a new value for global_step."
|
D | api_def_ResourceAccumulatorSetGlobalStep.pbtxt | 13 The new global_step value to set. 16 summary: "Updates the accumulator with a new value for global_step."
|