Home
last modified time | relevance | path

Searched refs:get_global_step (Results 1 – 25 of 54) sorted by relevance

123

/external/tensorflow/tensorflow/python/training/
Dtraining_util_test.py40 self.assertIsNone(training_util.get_global_step())
47 training_util.get_global_step)
49 training_util.get_global_step, g)
53 self.assertIsNone(training_util.get_global_step())
60 training_util.get_global_step)
62 training_util.get_global_step, g)
65 self.assertIsNone(training_util.get_global_step())
77 self.assertIsNone(training_util.get_global_step())
84 training_util.get_global_step(), expected_dtype=dtypes.int32)
86 training_util.get_global_step(g), expected_dtype=dtypes.int32)
[all …]
Dtraining_util.py73 def get_global_step(graph=None): function
122 if get_global_step(graph) is not None:
160 global_step_tensor = get_global_step(graph)
228 global_step_tensor = get_global_step(graph)
246 global_step_tensor = get_global_step(graph)
/external/tensorflow/tensorflow/contrib/learn/python/learn/
Dgraph_actions_test.py508 train_op = state_ops.assign_add(variables_lib.get_global_step(), 1)
527 train_op = state_ops.assign_add(variables_lib.get_global_step(), 1)
535 self._output_dir, variables_lib.get_global_step().name)
540 train_op = state_ops.assign_add(variables_lib.get_global_step(), 1)
548 self._output_dir, variables_lib.get_global_step().name)
554 train_op = state_ops.assign_add(variables_lib.get_global_step(), 1)
562 self._output_dir, variables_lib.get_global_step().name)
567 train_op = state_ops.assign_add(variables_lib.get_global_step(), 1)
575 self._output_dir, variables_lib.get_global_step().name)
583 state_ops.assign_add(variables_lib.get_global_step(), 1),
[all …]
/external/tensorflow/tensorflow/contrib/opt/python/training/
Ddrop_stale_gradient_optimizer_test.py122 global_step = training_util.get_global_step(graphs[0])
145 global_step = training_util.get_global_step(graphs[0])
168 global_step = training_util.get_global_step(graphs[0])
199 global_step = training_util.get_global_step(graphs[0])
230 global_step = training_util.get_global_step(graphs[0])
265 global_step = training_util.get_global_step(graphs[0])
Ddrop_stale_gradient_optimizer.py66 self._local_step = training_util.get_global_step() + 0
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Dtrainer_hooks.py57 self._global_step_tensor = training_util.get_global_step()
164 self._global_step_tensor = training_util.get_global_step()
217 self._global_step_tensor = training_util.get_global_step()
Dmodel.py108 global_step = training_util.get_global_step()
266 global_step = training_util.get_global_step()
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Destimators_test.py65 update_global_step = training_util.get_global_step().assign_add(1)
105 update_global_step = training_util.get_global_step().assign_add(1)
146 update_global_step = training_util.get_global_step().assign_add(1)
Destimator_input_test.py116 training_util.get_global_step(),
132 training_util.get_global_step(),
145 training_util.get_global_step(),
159 training_util.get_global_step(),
Destimator_test.py135 training_util.get_global_step(),
151 training_util.get_global_step(),
164 training_util.get_global_step(),
178 training_util.get_global_step(),
251 update_global_step = training_util.get_global_step().assign_add(1)
316 train_op=training_util.get_global_step().assign_add(1))
406 training_util.get_global_step().assign_add(1))
419 update_global_step = training_util.get_global_step().assign_add(1)
434 update_global_step = training_util.get_global_step().assign_add(1)
453 update_global_step = training_util.get_global_step().assign_add(1)
[all …]
Dlogistic_regressor_test.py61 training_util.get_global_step(),
/external/tensorflow/tensorflow/contrib/framework/python/ops/
Dvariables_test.py194 self.assertEquals(None, variables_lib2.get_global_step())
201 variables_lib2.get_global_step)
203 variables_lib2.get_global_step, g)
207 self.assertEquals(None, variables_lib2.get_global_step())
214 variables_lib2.get_global_step)
216 variables_lib2.get_global_step, g)
219 self.assertEquals(None, variables_lib2.get_global_step())
231 self.assertEquals(None, variables_lib2.get_global_step())
238 variables_lib2.get_global_step(), expected_dtype=dtypes.int32)
240 variables_lib2.get_global_step(g), expected_dtype=dtypes.int32)
[all …]
Dvariables.py117 global_step_tensor = get_global_step(graph)
123 def get_global_step(graph=None): function
124 return training_util.get_global_step(graph)
/external/tensorflow/tensorflow/contrib/tpu/
DREADME.md24 train_op = optimizer.minimize(loss, global_step=tf.train.get_global_step())
/external/tensorflow/tensorflow/python/tpu/
Dsession_support.py321 if training_util.get_global_step() is None and self.saver() is not None:
382 global_step=training_util.get_global_step(),
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dquantization_mnist_test.py51 from tensorflow.python.training.training_util import get_global_step
240 train_op = optimizer.minimize(loss, global_step=get_global_step())
/external/tensorflow/tensorflow/contrib/distribute/python/examples/
Dsimple_estimator_example.py57 global_step = tf.train.get_global_step()
/external/tensorflow/tensorflow/examples/learn/
Diris_custom_model.py56 train_op = optimizer.minimize(loss, global_step=tf.train.get_global_step())
Diris_custom_decay_dnn.py54 global_step = tf.train.get_global_step()
/external/tensorflow/tensorflow/examples/get_started/regression/
Dcustom_regression.py63 loss=average_loss, global_step=tf.train.get_global_step())
/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model_test.py386 sess.run(training_module.get_global_step()))
396 sess.run(training_module.get_global_step()))
408 sess.run(training_module.get_global_step()))
/external/tensorflow/tensorflow/examples/tutorials/layers/
Dcnn_mnist.py107 global_step=tf.train.get_global_step())
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dgmm.py167 incr_step = state_ops.assign_add(training_util.get_global_step(), 1)
/external/tensorflow/tensorflow/contrib/tensor_forest/client/
Drandom_forest.py90 {'global_step': training_util.get_global_step(),
228 state_ops.assign_add(training_util.get_global_step(), 1))
/external/tensorflow/tensorflow/contrib/autograph/examples/benchmarks/
Dcartpole_benchmark.py178 zip(grad_list, var_list), global_step=tf.train.get_global_step())
360 zip(grad_list, var_list), global_step=tf.train.get_global_step())

123