Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/training/
Dtraining_util_test.py38 self.assertIsNone(training_util.get_global_step())
46 training_util.get_global_step)
48 training_util.get_global_step, g)
52 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())
85 training_util.get_global_step(), expected_dtype=dtypes.int32)
87 training_util.get_global_step(g), expected_dtype=dtypes.int32)
[all …]
Dtraining_util.py72 def get_global_step(graph=None): function
121 if get_global_step(graph) is not None:
159 global_step_tensor = get_global_step(graph)
226 global_step_tensor = get_global_step(graph)
244 global_step_tensor = get_global_step(graph)
Dtraining.py107 from tensorflow.python.training.training_util import get_global_step
Dbasic_session_run_hooks.py363 self._global_step_tensor = training_util.get_global_step()
690 self._summary_tag = training_util.get_global_step().op.name + "/sec"
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dquantization_mnist_test.py51 from tensorflow.python.training.training_util import get_global_step
235 train_op = optimizer.minimize(loss, global_step=get_global_step())
/external/tensorflow/tensorflow/python/tpu/
Dsession_support.py338 if training_util.get_global_step() is None and self.saver() is not None:
399 global_step=training_util.get_global_step(),
Dasync_checkpoint_test.py75 train_op = optimizer.minimize(loss, training.get_global_step())
/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model_experimental_test.py395 sess.run(training_module.get_global_step()))
405 sess.run(training_module.get_global_step()))
417 sess.run(training_module.get_global_step()))
/external/tensorflow/tensorflow/python/distribute/
Dparameter_server_strategy_test.py848 get_step = training_util.get_global_step()
862 get_step = training_util.get_global_step()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.pbtxt328 name: "get_global_step"
/external/tensorflow/tensorflow/python/keras/
Doptimizer_v1.py803 global_step = training_util.get_global_step()
/external/tensorflow/
DRELEASE.md1778 `tf.global_variables_initializer` and `tf.get_global_step`.