Home
last modified time | relevance | path

Searched refs:global_step (Results 1 – 25 of 125) sorted by relevance

12345

/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Dlegacy_learning_rate_decay.py33 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/
Dadagrad_da_test.py37 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/
Dsync_replicas_optimizer_test.py44 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 …]
Dadagrad_da_test.py40 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 …]
Dsupervisor.py220 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 …]
Dbasic_session_run_hooks.py368 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 …]
Dadagrad_da.py50 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,
Dsync_replicas_optimizer.py224 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 …]
Dbasic_session_run_hooks_test.py72 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 …]
Dtraining_util_test.py31 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)
Doptimizer_test.py54 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])
Doptimizer.py363 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/
Dfake_summary_writer.py112 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)
Dwriter.py101 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/
Ddebug_utils.py32 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/
Dsummary_file_writer.cc70 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/
Dsummary_interface.h38 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/
Dtracking_util_with_v1_optimizers_test.py89 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/
Dasync_checkpoint.py107 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/
Dloss_scale_optimizer.py156 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/
Dtensorflow.summary.-file-writer.pbtxt16 …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=[\'…
Dtensorflow.train.-adagrad-d-a-optimizer.pbtxt21 …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/
Diterator_ops.py279 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/
Dapi_def_AccumulatorSetGlobalStep.pbtxt12 The new global_step value to set.
15 summary: "Updates the accumulator with a new value for global_step."
Dapi_def_ResourceAccumulatorSetGlobalStep.pbtxt13 The new global_step value to set.
16 summary: "Updates the accumulator with a new value for global_step."

12345