Searched defs:global_step (Results 1 - 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/core/debug/ |
H A D | debugger_state_impl.cc | 39 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 D | debug_io_utils.cc | 325 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 D | training_util.py | 44 @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 D | 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 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 D | training.py | 86 @@global_step 185 from tensorflow.python.training.training_util import global_step namespace
|
/external/tensorflow/tensorflow/core/kernels/ |
H A D | training_ops.cc | 117 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