Searched refs:_input_fn (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | checkpoint_input_pipeline_hook_test.py | 81 def _input_fn(): function
86 est.train(_input_fn, steps=2, hooks=[self._build_iterator_saver_hook(est)])
88 est.train(_input_fn, steps=2, hooks=[self._build_iterator_saver_hook(est)])
94 def _input_fn(): function
101 est.train(_input_fn, steps=2, hooks=[self._build_iterator_saver_hook(est)])
103 est.train(_input_fn, steps=2, hooks=[self._build_iterator_saver_hook(est)])
109 def _input_fn(): function
114 est.train(_input_fn, steps=2, hooks=[self._build_iterator_saver_hook(est)])
116 est.train(_input_fn, steps=2, hooks=[self._build_iterator_saver_hook(est)])
119 est.train(_input_fn, steps=2)
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_test_lib.py | 322 def _input_fn(input_context): function
338 return _input_fn
|