Home
last modified time | relevance | path

Searched defs:global_step (Results 1 – 25 of 31) sorted by relevance

12

/aosp_15_r20/external/pytorch/torch/utils/tensorboard/
H A Dwriter.py101 def add_summary(self, summary, global_step=None, walltime=None): argument
293 global_step=None, argument
346 global_step=None, argument
383 def add_scalars(self, main_tag, tag_scalar_dict, global_step=None, walltime=None): argument
431 global_step=None, argument
463 global_step=None, argument
512 global_step=None, argument
578 self, tag, img_tensor, global_step=None, walltime=None, dataformats="CHW" argument
629 self, tag, img_tensor, global_step=None, walltime=None, dataformats="NCHW" argument
677 global_step=None, argument
[all …]
H A D_embedding.py64 def get_embedding_info(metadata, label_img, subdir, global_step, tag): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/optimizer_v2/
H A Dlegacy_learning_rate_decay.py30 global_step, argument
184 global_step, argument
285 global_step, argument
373 global_step, argument
455 def cosine_decay(learning_rate, global_step, decay_steps, alpha=0.0, name=None): argument
519 global_step, argument
596 global_step, argument
681 global_step, argument
/aosp_15_r20/external/tensorflow/tensorflow/python/summary/writer/
H A Dwriter.py97 def add_summary(self, summary, global_step=None): argument
140 def add_session_log(self, session_log, global_step=None): argument
154 def _add_graph_def(self, graph_def, global_step=None): argument
159 def add_graph(self, graph, global_step=None, graph_def=None): argument
225 def add_meta_graph(self, meta_graph_def, global_step=None): argument
247 def add_run_metadata(self, run_metadata, tag, global_step=None): argument
H A Dfake_summary_writer.py108 def add_graph(self, graph, global_step=None, graph_def=None): argument
116 def add_meta_graph(self, meta_graph_def, global_step=None): argument
123 def add_session_log(self, session_log, global_step=None): argument
127 def add_run_metadata(self, run_metadata, tag, global_step=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/summary/
H A Dsummary_file_writer.cc80 Status WriteTensor(int64_t global_step, Tensor t, const string& tag, in WriteTensor()
103 Status WriteScalar(int64_t global_step, Tensor t, in WriteScalar()
113 Status WriteHistogram(int64_t global_step, Tensor t, in WriteHistogram()
123 Status WriteImage(int64_t global_step, Tensor t, const string& tag, in WriteImage()
133 Status WriteAudio(int64_t global_step, Tensor t, const string& tag, in WriteAudio()
143 Status WriteGraph(int64_t global_step, in WriteGraph()
H A Dsummary_db_writer.cc919 Status WriteTensor(int64_t global_step, Tensor t, const string& tag, in WriteTensor()
929 Status WriteScalar(int64_t global_step, Tensor t, in WriteScalar()
937 Status WriteGraph(int64_t global_step, std::unique_ptr<GraphDef> g) override { in WriteGraph()
946 Status WriteHistogram(int64_t global_step, Tensor t, in WriteHistogram()
957 Status WriteImage(int64_t global_step, Tensor t, const string& tag, in WriteImage()
968 Status WriteAudio(int64_t global_step, Tensor t, const string& tag, in WriteAudio()
/aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/
H A Ddebug_utils.py27 global_step=-1): argument
85 global_step=-1, argument
205 global_step=-1, argument
H A Dsession_debug_testlib.py126 global_step=-1, argument
/aosp_15_r20/external/tensorflow/tensorflow/python/training/experimental/
H A Dloss_scale_optimizer.py153 def apply_gradients(self, grads_and_vars, global_step=None, name=None): argument
191 global_step=None, argument
225 def _apply_gradients(self, distribution, grads_and_vars, global_step, name): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dsupervisor.py216 global_step=USE_DEFAULT, argument
482 def _init_global_step(self, global_step=USE_DEFAULT): argument
587 def global_step(self): member in Supervisor
881 def summary_computed(self, sess, summary, global_step=None): argument
H A Doptimizer.py430 def minimize(self, loss, global_step=None, var_list=None, argument
623 def apply_gradients(self, grads_and_vars, global_step=None, name=None): argument
746 global_step=None, argument
H A Dbasic_session_run_hooks_test.py68 def before_save(self, session, global_step): argument
71 def after_save(self, session, global_step): argument
76 def end(self, session, global_step): argument
H A Dtraining_util.py37 def global_step(sess, global_step_tensor): function
H A Dtraining_util_test.py27 def _assert_global_step(self, global_step, expected_dtype=dtypes.int64): argument
H A Dadagrad_da.py46 global_step, argument
H A Dbasic_session_run_hooks.py371 def _update_steps_per_run_variable(self, global_step, session): argument
709 def _log_and_record(self, elapsed_steps, elapsed_time, global_step): argument
H A Dsync_replicas_optimizer.py221 def apply_gradients(self, grads_and_vars, global_step=None, name=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Ddebug.proto65 int64 global_step = 10; field
/aosp_15_r20/external/tensorflow/tensorflow/core/debug/
H A Ddebugger_state_impl.cc42 const int64_t global_step, const int64_t session_run_index, in PublishDebugMetadata()
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu_optimizer.py164 def apply_gradients(self, grads_and_vars, global_step=None, name=None): argument
/aosp_15_r20/external/pytorch/torch/distributed/checkpoint/
H A Dstate_dict_loader.py216 def global_step(all_local_plans): function
H A Dstate_dict_saver.py308 def global_step(all_local_plans): function
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/
H A Diterator_ops.py313 global_step=None, argument
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dsummary_ops_v2.py184 def record_summaries_every_n_global_steps(n, global_step=None): argument

12