Home
last modified time | relevance | path

Searched defs:input_fn (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dasync_checkpoint_test.py56 def input_fn(params): function
/external/tensorflow/tensorflow/python/ops/
Daccumulate_n_benchmark.py120 def _RunBenchmark(self, tag, input_fn, sizes, ninputs, repeats): argument
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_v2_utils.py95 def execution_function(input_fn): argument
Dtraining_generator_test.py525 def test_convert_to_generator_like(self, input_fn, inputs): argument
/external/tensorflow/tensorflow/python/keras/distribute/
Ddistributed_training_utils.py860 def distributed_function(input_fn): argument
876 def execution_function(input_fn): argument
Dkeras_correctness_test_base.py250 input_fn, argument
/external/tensorflow/tensorflow/python/distribute/
Dcollective_all_reduce_strategy.py427 def _make_input_fn_iterator( argument
Dminimize_loss_test.py66 def _get_iterator(self, strategy, input_fn): argument
Ddistribute_lib.py572 def make_input_fn_iterator(self, argument
1050 def make_input_fn_iterator(self, # pylint: disable=useless-super-delegation argument
1493 def _make_input_fn_iterator(self, input_fn, replication_mode): argument
2205 def _make_input_fn_iterator(self, argument
Dparameter_server_strategy_test.py515 def _test_input_fn_iterator(self, argument
Done_device_strategy.py278 def _make_input_fn_iterator( argument
Dcollective_all_reduce_strategy_test.py345 def _test_input_fn_iterator(self, argument
Dparameter_server_strategy.py327 def _make_input_fn_iterator( argument
Ddistribute_lib_test.py84 def _make_input_fn_iterator( argument
Dtpu_strategy.py338 def _make_input_fn_iterator( argument
Dmirrored_strategy.py623 def _make_input_fn_iterator( argument
Dstrategy_test_lib.py310 def _test_input_fn_iterable( argument
Dinput_lib_test.py217 def _create_dataset_or_input_fn(self, input_type, input_fn): argument
Dinput_lib.py697 def __init__(self, input_fn, input_workers, input_contexts, strategy): argument
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py1091 def build(self, input_fn): argument