Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/coordinator/
Dcluster_coordinator_test.py479 def input_fn(): function
505 def input_fn(): function
549 def input_fn(): function
569 def input_fn(): function
582 def input_fn(): function
665 def input_fn(): function
684 def input_fn(): function
/external/tensorflow/tensorflow/python/tpu/
Dasync_checkpoint_test.py56 def input_fn(params): function
Dtpu_embedding_v2_test.py1101 def input_fn(ctx): function
1159 def input_fn(ctx): function
Dtpu_embedding_v2_correctness_test.py268 def input_fn(ctx): function
/external/tensorflow/tensorflow/core/platform/
Dram_file_system_test.py115 def input_fn(): 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/distribute/
Ddistributed_training_utils_v1.py847 def distributed_function(input_fn): argument
862 def execution_function(input_fn): argument
Dminimize_loss_test.py69 def _get_iterator(self, strategy, input_fn): argument
Dkeras_correctness_test_base.py255 input_fn, argument
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_generator_test.py521 def test_convert_to_generator_like(self, input_fn, inputs): argument
/external/tensorflow/tensorflow/python/distribute/
Ddistribute_lib.py930 def make_input_fn_iterator(self, argument
1846 def make_input_fn_iterator(self, # pylint: disable=useless-super-delegation argument
2207 def _make_input_fn_iterator(self, input_fn, replication_mode): argument
3437 def _make_input_fn_iterator(self, argument
Dcollective_all_reduce_strategy_test.py213 def _test_input_fn_iterator(self, argument
Dcollective_all_reduce_strategy.py645 def _make_input_fn_iterator( argument
Dparameter_server_strategy_test.py522 def _test_input_fn_iterator(self, argument
Done_device_strategy.py296 def _make_input_fn_iterator( argument
Dparameter_server_strategy.py355 def _make_input_fn_iterator( argument
Dmirrored_strategy.py544 def _make_input_fn_iterator( argument
Ddistribute_lib_test.py83 def _make_input_fn_iterator( argument
Dstrategy_test_lib.py340 def _test_input_fn_iterable( argument
Dtpu_strategy.py887 def _make_input_fn_iterator( argument
Dinput_lib_test.py276 def _create_dataset_or_input_fn(self, input_type, input_fn): argument
Dinput_lib.py1375 def __init__(self, input_fn, input_workers, input_contexts, strategy): argument
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py1090 def build(self, input_fn): argument