Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/coordinator/
Dcluster_coordinator_test.py504 def input_fn(): function
532 def input_fn(): function
577 def input_fn(): function
602 def input_fn(): function
616 def input_fn(): function
708 def input_fn(): function
769 def input_fn(): function
982 def input_fn(): function
1314 def input_fn(): function
1364 def input_fn(): function
/external/tensorflow/tensorflow/python/data/util/
Dsparse_test.py316 def testSerializeDeserialize(self, input_fn): argument
332 def testSerializeManyDeserialize(self, input_fn): argument
Drandom_seed_test.py122 def testRandomSeed(self, input_fn, output_fn): argument
/external/tensorflow/tensorflow/python/tpu/
Dasync_checkpoint_test.py56 def input_fn(params): function
Dtpu_embedding_v2_test.py1121 def input_fn(ctx): function
1179 def input_fn(ctx): function
Dtpu_embedding_v2_correctness_test.py266 def input_fn(ctx): function
/external/tensorflow/tensorflow/core/platform/
Dram_file_system_test.py130 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.py837 def distributed_function(input_fn): argument
852 def execution_function(input_fn): argument
Dminimize_loss_test.py65 def _get_iterator(self, strategy, input_fn): argument
Dkeras_correctness_test_base.py251 input_fn, argument
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_generator_test.py517 def test_convert_to_generator_like(self, input_fn, inputs): argument
/external/tensorflow/tensorflow/python/distribute/
Dcollective_all_reduce_strategy_test.py214 def _test_input_fn_iterator(self, argument
Ddistribute_lib.py966 def make_input_fn_iterator(self, argument
1896 def make_input_fn_iterator(self, # pylint: disable=useless-super-delegation argument
2277 def _make_input_fn_iterator(self, input_fn, replication_mode): argument
3661 def _make_input_fn_iterator(self, argument
Dparameter_server_strategy_test.py522 def _test_input_fn_iterator(self, argument
Done_device_strategy.py296 def _make_input_fn_iterator( argument
Dcollective_all_reduce_strategy.py665 def _make_input_fn_iterator( argument
Dparameter_server_strategy.py356 def _make_input_fn_iterator( argument
Ddistribute_lib_test.py83 def _make_input_fn_iterator( argument
Dmirrored_strategy.py543 def _make_input_fn_iterator( argument
Dstrategy_test_lib.py340 def _test_input_fn_iterable( argument
Dtpu_strategy.py886 def _make_input_fn_iterator( argument
Dinput_lib_test.py274 def _create_dataset_or_input_fn(self, input_type, input_fn): argument
Dinput_lib.py1769 def __init__(self, input_fn, input_workers, input_contexts, strategy): argument
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py1176 def build(self, input_fn): argument