Searched defs:global_step (Results 1 - 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/debug/
H A Ddebugger_state_impl.cc39 const int64 global_step, const int64 session_run_index,
43 return DebugIO::PublishDebugMetadata(global_step, session_run_index,
38 PublishDebugMetadata( const int64 global_step, const int64 session_run_index, const int64 executor_step_index, const std::vector<string>& input_names, const std::vector<string>& output_names, const std::vector<string>& target_names) argument
H A Ddebug_io_utils.cc325 const int64 global_step, const int64 session_run_index,
334 oss << "\"global_step\":" << global_step << ","; local
324 PublishDebugMetadata( const int64 global_step, const int64 session_run_index, const int64 executor_step_index, const std::vector<string>& input_names, const std::vector<string>& output_names, const std::vector<string>& target_nodes, const std::unordered_set<string>& debug_urls) argument
/external/tensorflow/tensorflow/python/training/
H A Dtraining_util.py44 @tf_export('train.global_step')
45 def global_step(sess, global_step_tensor): function
49 # Creates a variable to hold the global_step.
50 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`.
96 global_step_tensor = graph.get_tensor_by_name('global_step:0')
100 logging.error('Multiple tensors in global_step collectio
[all...]
H A Dsupervisor.py203 # 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
320 self._init_global_step(global_step=global_step)
485 def _init_global_step(self, global_step=USE_DEFAULT):
486 """Initializes global_step.
489 global_step
590 def global_step(self): member in class:Supervisor
[all...]
H A Dtraining.py86 @@global_step
185 from tensorflow.python.training.training_util import global_step namespace
/external/tensorflow/tensorflow/core/kernels/
H A Dtraining_ops.cc117 typename TTypes<T>::ConstScalar lr, int64 global_step,
134 var.constant(static_cast<float>(global_step)) * var.constant(l1()))
137 var.constant(static_cast<float>(global_step) * lr()) +
143 var.constant(static_cast<float>(global_step) * lr()) +
1608 const Tensor& global_step = ctx->input(7); variable
1609 OP_REQUIRES(ctx, IsLegacyScalar(global_step.shape()),
1610 errors::InvalidArgument("global_step is not a scalar: ",
1611 global_step.shape().DebugString()));
1617 global_step.scalar<int64>()(), l1.scalar<T>(), l2.scalar<T>(),
1713 const Tensor& global_step variable
114 operator ()(const CPUDevice& d, typename TTypes<T>::Flat var, typename TTypes<T>::Flat gradient_accum, typename TTypes<T>::Flat gradient_squared_accum, typename TTypes<T>::ConstScalar lr, int64 global_step, typename TTypes<T>::ConstScalar l1, typename TTypes<T>::ConstScalar l2, typename TTypes<T>::ConstFlat grad) argument
[all...]

Completed in 243 milliseconds