Searched refs:_input_fn (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| H A D | checkpoint_input_pipeline_hook_test.py | 73 def _input_fn(): function
78 est.train(_input_fn, steps=2, hooks=[self._build_iterator_saver_hook(est)])
80 est.train(_input_fn, steps=2, hooks=[self._build_iterator_saver_hook(est)])
86 def _input_fn(): function
93 est.train(_input_fn, steps=2, hooks=[self._build_iterator_saver_hook(est)])
95 est.train(_input_fn, steps=2, hooks=[self._build_iterator_saver_hook(est)])
101 def _input_fn(): function
106 est.train(_input_fn, steps=2, hooks=[self._build_iterator_saver_hook(est)])
108 est.train(_input_fn, steps=2, hooks=[self._build_iterator_saver_hook(est)])
111 est.train(_input_fn, steps=2)
[all …]
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/experimental/ |
| H A D | _attention.py | 661 input_fn=self._input_fn, # type: ignore[arg-type]
666 def _input_fn( member in _AttentionContextParallel
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | strategy_test_lib.py | 339 def _input_fn(input_context): function
355 return _input_fn
|