| /external/tensorflow/tensorflow/python/training/ |
| training_util.py | 42 @tf_export(v1=['train.global_step'])
43 def global_step(sess, global_step_tensor): function
47 # Create a variable to hold the global_step.
48 global_step_tensor = tf.Variable(10, trainable=False, name='global_step')
54 print('global_step: %s' % tf.train.global_step(sess, global_step_tensor))
56 global_step: 10
77 in the collection `GLOBAL_STEP`, or by name `global_step:0`.
91 global_step_tensors = graph.get_collection(ops.GraphKeys.GLOBAL_STEP)
[all...] |
| supervisor.py | 203 # and 'global_step' parameters of Supervisor.__init__() to indicate that
220 global_step=USE_DEFAULT,
272 global_step: An integer Tensor of size 1 that counts steps. The value
273 from 'global_step' is used in summaries and checkpoint filenames.
274 Default to the op named 'global_step' in the graph if it exists, is of
322 self._init_global_step(global_step=global_step)
489 def _init_global_step(self, global_step=USE_DEFAULT):
490 """Initializes global_step.
493 global_step: An integer Tensor of size 1 that counts steps. I
594 def global_step(self): member in class:Supervisor
[all...] |
| /external/tensorflow/tensorflow/python/grappler/ |
| hierarchical_controller.py | 370 math_ops.less(self.global_step,
390 math_ops.less(self.global_step,
408 self.global_step,
425 def global_step(self): member in class:HierarchicalController
570 self.global_step > self.hparams.stop_noise_step, lambda: logits,
716 self.global_step < self.hparams.stop_sampling,
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| training_ops.cc | 381 xla::XlaOp global_step = variable
387 xla::XlaOp denominator = global_step * lr * l2 + xla::Sqrt(squared_accum);
390 xla::Max(xla::Abs(accum) - global_step * l1, zero) /
[all...] |
| /external/tensorflow/tensorflow/core/debug/ |
| debug_io_utils.cc | 328 const int64 global_step, const int64 session_run_index,
337 oss << "\"global_step\":" << global_step << ","; local
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| training_ops.cc | 118 typename TTypes<T>::ConstScalar lr, int64 global_step,
135 var.constant(static_cast<float>(global_step)) * var.constant(l1()))
138 var.constant(static_cast<float>(global_step) * lr()) +
144 var.constant(static_cast<float>(global_step) * lr()) +
1705 const Tensor& global_step = ctx->input(7); variable
1811 const Tensor& global_step = ctx->input(8); variable
[all...] |
