Home
last modified time | relevance | path

Searched refs:_FakeHook (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/estimator/
Dmodel_fn_test.py34 class _FakeHook(session_run_hook.SessionRunHook): class
72 training_chief_hooks=[_FakeHook()],
73 training_hooks=[_FakeHook()],
75 evaluation_hooks=[_FakeHook()],
76 prediction_hooks=[_FakeHook()])
224 training_chief_hooks=[_FakeHook()],
225 training_hooks=[_FakeHook()],
227 evaluation_hooks=[_FakeHook()])
466 training_chief_hooks=[_FakeHook()],
467 training_hooks=[_FakeHook()],
[all …]
Dtraining_test.py155 class _FakeHook(session_run_hook.SessionRunHook): class
196 hooks = [_FakeHook()]
231 hooks = [_FakeHook()]
286 _FakeHook()])
290 training.EvalSpec(input_fn=lambda: 1, exporters=_FakeHook())
436 input_fn=lambda: 1, max_steps=2, hooks=[_FakeHook()])
466 input_fn=lambda: 1, max_steps=2, hooks=[_FakeHook()])
468 extra_hooks = [_FakeHook()]
663 input_fn=lambda: 1, max_steps=2, hooks=[_FakeHook()])
693 input_fn=lambda: 1, max_steps=2, hooks=[_FakeHook()])
[all …]