Home
last modified time | relevance | path

Searched refs:run_context (Results 1 – 25 of 51) sorted by relevance

123

/external/tensorflow/tensorflow/python/debug/wrappers/
Dhooks.py91 def before_run(self, run_context): argument
94 run_context.session,
112 run_context.original_args.fetches, run_context.original_args.feed_dict,
143 def after_run(self, run_context, run_values): argument
185 def before_run(self, run_context): argument
189 run_context.session,
200 run_context.original_args.fetches, run_context.original_args.feed_dict)
205 run_context.session.graph,
219 def after_run(self, run_context, run_values): argument
266 def before_run(self, run_context): argument
[all …]
/external/tensorflow/tensorflow/python/training/
Dbasic_session_run_hooks.py242 def before_run(self, run_context): # pylint: disable=unused-argument argument
265 def after_run(self, run_context, run_values): argument
266 _ = run_context
379 def after_run(self, run_context, run_values): argument
382 global_step = run_context.session.run(self._global_step_tensor)
384 run_context.request_stop()
386 self._update_steps_per_run_variable(global_step, run_context.session)
429 def before_run(self, run_context): # pylint: disable=unused-argument argument
432 def after_run(self, run_context, run_values): argument
440 step = run_context.session.run(self._global_step_tensor)
[all …]
Devaluation.py111 def before_run(self, run_context): argument
115 def after_run(self, run_context, run_values): argument
123 self._steps_per_run_variable.load(steps, session=run_context.session)
130 run_context.request_stop()
155 def before_run(self, run_context): argument
159 def after_run(self, run_context, run_values): argument
169 run_context.request_stop()
Dsession_run_hook.py129 def before_run(self, run_context): # pylint: disable=unused-argument argument
153 run_context, # pylint: disable=unused-argument argument
/external/tensorflow/tensorflow/python/data/experimental/ops/
Diterator_ops.py284 def before_run(self, run_context): argument
286 self._restore_or_save_initial_ckpt(run_context.session)
288 return self._checkpoint_saver_hook.before_run(run_context)
290 def after_run(self, run_context, run_values): argument
291 self._checkpoint_saver_hook.after_run(run_context, run_values)
/external/tensorflow/tensorflow/python/tpu/
Dasync_checkpoint.py130 def before_run(self, run_context: Any): # pylint: disable=unused-argument
133 def after_run(self, run_context: session_run_hook.SessionRunContext,
135 global_step = run_context.session.run(self._global_step_tensor)
139 if self._save(run_context.session, global_step):
140 run_context.request_stop()
Dsession_support.py383 def after_run(self, run_context, run_values): argument
397 run_context.session,
406 fn(run_context, self._workers, lame_workers)
419 def __call__(self, run_context, all_workers, lame_workers): argument
420 del run_context, lame_workers
437 def __call__(self, run_context, all_workers, lame_workers): argument
451 def __call__(self, run_context, all_workers, lame_workers): argument
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.-session-run-hook.pbtxt14 …argspec: "args=[\'self\', \'run_context\', \'run_values\'], varargs=None, keywords=None, defaults=…
18 argspec: "args=[\'self\', \'run_context\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.estimator.-session-run-hook.pbtxt14 …argspec: "args=[\'self\', \'run_context\', \'run_values\'], varargs=None, keywords=None, defaults=…
18 argspec: "args=[\'self\', \'run_context\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.estimator.-global-step-waiter-hook.pbtxt16 …argspec: "args=[\'self\', \'run_context\', \'run_values\'], varargs=None, keywords=None, defaults=…
20 argspec: "args=[\'self\', \'run_context\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.estimator.-stop-at-step-hook.pbtxt16 …argspec: "args=[\'self\', \'run_context\', \'run_values\'], varargs=None, keywords=None, defaults=…
20 argspec: "args=[\'self\', \'run_context\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.estimator.-nan-tensor-hook.pbtxt16 …argspec: "args=[\'self\', \'run_context\', \'run_values\'], varargs=None, keywords=None, defaults=…
20 argspec: "args=[\'self\', \'run_context\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.train.-global-step-waiter-hook.pbtxt16 …argspec: "args=[\'self\', \'run_context\', \'run_values\'], varargs=None, keywords=None, defaults=…
20 argspec: "args=[\'self\', \'run_context\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.train.-feed-fn-hook.pbtxt16 …argspec: "args=[\'self\', \'run_context\', \'run_values\'], varargs=None, keywords=None, defaults=…
20 argspec: "args=[\'self\', \'run_context\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.train.-stop-at-step-hook.pbtxt16 …argspec: "args=[\'self\', \'run_context\', \'run_values\'], varargs=None, keywords=None, defaults=…
20 argspec: "args=[\'self\', \'run_context\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.train.-nan-tensor-hook.pbtxt16 …argspec: "args=[\'self\', \'run_context\', \'run_values\'], varargs=None, keywords=None, defaults=…
20 argspec: "args=[\'self\', \'run_context\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.estimator.-feed-fn-hook.pbtxt16 …argspec: "args=[\'self\', \'run_context\', \'run_values\'], varargs=None, keywords=None, defaults=…
20 argspec: "args=[\'self\', \'run_context\'], varargs=None, keywords=None, defaults=None"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.estimator.-session-run-hook.pbtxt14 …argspec: "args=[\'self\', \'run_context\', \'run_values\'], varargs=None, keywords=None, defaults=…
18 argspec: "args=[\'self\', \'run_context\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.estimator.-global-step-waiter-hook.pbtxt16 …argspec: "args=[\'self\', \'run_context\', \'run_values\'], varargs=None, keywords=None, defaults=…
20 argspec: "args=[\'self\', \'run_context\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.estimator.-feed-fn-hook.pbtxt16 …argspec: "args=[\'self\', \'run_context\', \'run_values\'], varargs=None, keywords=None, defaults=…
20 argspec: "args=[\'self\', \'run_context\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.estimator.-nan-tensor-hook.pbtxt16 …argspec: "args=[\'self\', \'run_context\', \'run_values\'], varargs=None, keywords=None, defaults=…
20 argspec: "args=[\'self\', \'run_context\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.estimator.-stop-at-step-hook.pbtxt16 …argspec: "args=[\'self\', \'run_context\', \'run_values\'], varargs=None, keywords=None, defaults=…
20 argspec: "args=[\'self\', \'run_context\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.estimator.experimental.-in-memory-evaluator-hook.pbtxt16 …argspec: "args=[\'self\', \'run_context\', \'run_values\'], varargs=None, keywords=None, defaults=…
20 argspec: "args=[\'self\', \'run_context\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.estimator.-summary-saver-hook.pbtxt16 …argspec: "args=[\'self\', \'run_context\', \'run_values\'], varargs=None, keywords=None, defaults=…
20 argspec: "args=[\'self\', \'run_context\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.estimator.-logging-tensor-hook.pbtxt16 …argspec: "args=[\'self\', \'run_context\', \'run_values\'], varargs=None, keywords=None, defaults=…
20 argspec: "args=[\'self\', \'run_context\'], varargs=None, keywords=None, defaults=None"

123