Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/python/debug/wrappers/
Dhooks.py85 def before_run(self, run_context): argument
88 run_context.session,
105 run_context.original_args.fetches, run_context.original_args.feed_dict,
137 run_context.session.graph._finalized = False
141 run_context.session, run_context.original_args.fetches,
142 run_context.original_args.feed_dict) as node_stepper:
148 def after_run(self, run_context, run_values): argument
190 def before_run(self, run_context): argument
194 run_context.session,
205 run_context.original_args.fetches, run_context.original_args.feed_dict)
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Dtrainer_hooks.py69 def before_run(self, run_context): argument
70 del run_context # Unused by feature importance summary saver hook.
79 def after_run(self, run_context, run_values): argument
80 del run_context # Unused by feature importance summary saver hook.
146 def before_run(self, run_context): argument
147 del run_context # unused by FeedFnHook.
172 def before_run(self, run_context): argument
173 del run_context # unused by StopTrainingAfterNTrees.
179 def after_run(self, run_context, run_values): argument
192 run_context.session.run(self._override_global_step_op)
[all …]
/external/tensorflow/tensorflow/python/training/
Dbasic_session_run_hooks.py229 def before_run(self, run_context): # pylint: disable=unused-argument argument
252 def after_run(self, run_context, run_values): argument
253 _ = run_context
366 def after_run(self, run_context, run_values): argument
369 global_step = run_context.session.run(self._global_step_tensor)
371 run_context.request_stop()
373 self._update_steps_per_run_variable(global_step, run_context.session)
416 def before_run(self, run_context): # pylint: disable=unused-argument argument
419 def after_run(self, run_context, run_values): argument
427 step = run_context.session.run(self._global_step_tensor)
[all …]
Devaluation.py111 def before_run(self, run_context): argument
116 def after_run(self, run_context, run_values): argument
124 self._steps_per_run_variable.load(steps, session=run_context.session)
131 run_context.request_stop()
156 def before_run(self, run_context): argument
161 def after_run(self, run_context, run_values): argument
171 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.py228 def before_run(self, run_context): argument
230 self._restore_or_save_initial_ckpt(run_context.session)
232 return self._checkpoint_saver_hook.before_run(run_context)
234 def after_run(self, run_context, run_values): argument
235 self._checkpoint_saver_hook.after_run(run_context, run_values)
/external/tensorflow/tensorflow/python/tpu/
Dasync_checkpoint.py121 def before_run(self, run_context): # pylint: disable=unused-argument argument
124 def after_run(self, run_context, run_values): argument
125 global_step = run_context.session.run(self._global_step_tensor)
129 if self._save(run_context.session, global_step):
130 run_context.request_stop()
Dsession_support.py366 def after_run(self, run_context, run_values): argument
380 run_context.session,
389 fn(run_context, self._workers, lame_workers)
402 def __call__(self, run_context, all_workers, lame_workers): argument
403 del run_context, lame_workers
420 def __call__(self, run_context, all_workers, lame_workers): argument
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dwals.py72 def before_run(self, run_context): argument
74 sess = run_context.session
106 def before_run(self, run_context): argument
108 run_context.session.run(self._global_step_incr_op)
133 def before_run(self, run_context): argument
136 def after_run(self, run_context, run_values): argument
139 run_context.request_stop()
Dkmeans.py57 def before_run(self, run_context): argument
58 del run_context # unused
61 def after_run(self, run_context, run_values): argument
68 run_context.request_stop()
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dkmeans.py68 def before_run(self, run_context): argument
69 del run_context
73 def after_run(self, run_context, run_values): argument
80 run_context.request_stop()
/external/tensorflow/tensorflow/tools/api/golden/v1/
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.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.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.-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.-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.-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.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.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.-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.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"

123