| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | dataset_creator_test.py | 40 dataset_fn = lambda: 3 function
46 dataset_fn = lambda: dataset_ops.DatasetV2.from_tensor_slices([1, 1]) function
64 def dataset_fn(input_context): function
|
| D | dataset_creator.py | 55 def __init__(self, dataset_fn): argument
|
| D | kpl_test_utils.py | 87 def dataset_fn(self, feature_mapper, label_mapper): member in DistributeKplTestUtils
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib_test.py | 302 dataset_fn = lambda _: dataset_ops.DatasetV2.range(10) function
340 dataset_fn = lambda _: dataset_ops.DatasetV1.range(10) function
369 dataset_fn = lambda _: dataset_ops.Dataset.range(10) function
392 dataset_fn = lambda _: dataset_ops.DatasetV1.range(10) function
424 dataset_fn = lambda _: dataset_ops.Dataset.range(10) function
457 dataset_fn = lambda _: dataset_ops.Dataset.range(10) function
490 def dataset_fn(ctx): function
525 def dataset_fn(ctx): function
581 dataset_fn = lambda _: dataset_ops.Dataset.range(9).batch( # pylint: disable=g-long-lambda function
625 def dataset_fn(_): function
[all …]
|
| D | single_loss_example.py | 36 def dataset_fn(): function
57 def dataset_fn(): function
89 def dataset_fn(): function
|
| D | step_fn.py | 53 def __init__(self, dataset_fn, distribution): argument
88 def __init__(self, dataset_fn, loss_fn, optimizer, distribution, argument
|
| D | distribute_lib_test.py | 91 def _distribute_datasets_from_function(self, dataset_fn, options): argument
520 dataset_fn = lambda _: dataset_ops.DatasetV2.range(10).batch(2) function
525 dataset_fn = lambda _: dataset_ops.DatasetV1.range(10).batch(2) function
537 dataset_fn = lambda _: dataset_ops.DatasetV2.range(10).batch(2) function
544 dataset_fn = lambda _: dataset_ops.DatasetV2.range(10).batch(2) function
|
| D | tpu_strategy_test.py | 718 def dataset_fn(_): function
768 def dataset_fn(_): function
818 def dataset_fn(_): function
939 def dataset_fn(ctx): function
954 def dataset_fn(ctx): function
992 dataset_fn = lambda: dataset_ops.Dataset.range(10) function
1003 dataset_fn = lambda: dataset_ops.Dataset.range(10) function
|
| D | one_device_strategy_test.py | 58 dataset_fn = lambda: dataset_ops.Dataset.range(10) function
68 dataset_fn = lambda: dataset_ops.Dataset.range(10) function
|
| D | strategy_common_test.py | 200 def dataset_fn(input_context): function
244 def dataset_fn(input_context): function
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | stats_dataset_ops_test.py | 342 def dataset_fn(): function
355 def dataset_fn(): function
368 def dataset_fn(): function
384 def dataset_fn(): function
410 def dataset_fn(): function
|
| D | parallel_interleave_test.py | 59 def dataset_fn(self, input_values, cycle_length, block_length, sloppy, member in ParallelInterleaveTest
634 def dataset_fn(input_values, cycle_length, block_length, sloppy, function
678 def dataset_fn(input_values, cycle_length, block_length, sloppy, function
745 def dataset_fn(delay_ms): function
|
| D | dense_to_sparse_batch_test.py | 95 def dataset_fn(input_tensor): function
|
| D | scan_test.py | 57 def dataset_fn(start, step, take): function
95 def dataset_fn(start, step, take): function
|
| D | make_tf_record_dataset_test.py | 112 def dataset_fn(): function
|
| D | get_single_element_test.py | 90 def dataset_fn(): function
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | test_example.py | 34 def dataset_fn(): function
66 def dataset_fn(): function
|
| D | parameter_server_training_test.py | 136 def dataset_fn(): function
281 def dataset_fn(input_context): function
|
| D | keras_metrics_test.py | 93 def _test_metric(self, distribution, dataset_fn, metric_init_fn, expected_fn): argument
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | reduce_test.py | 144 def dataset_fn(): function
163 def dataset_fn(): function
183 def dataset_fn(): function
212 def dataset_fn(): function
|
| D | list_files_test.py | 94 def dataset_fn(): function
125 def dataset_fn(): function
|
| D | dataset_test.py | 140 def _testUnaryInputs(self, dataset_fn): argument
200 def _testInputsWithInterleaveFn(self, dataset_fn, interleave_parallelism): argument
223 def _testBinaryInputs(self, dataset_fn): argument
232 def _testVariadicInputs(self, dataset_fn, input_datasets): argument
|
| D | cache_test.py | 60 def dataset_fn(count=5, filename=None): function
255 dataset_fn = lambda: dataset_ops.Dataset.range(5).cache() function
|
| D | cardinality_test.py | 190 def testCardinality(self, dataset_fn, expected_result): argument
|
| D | text_line_dataset_test.py | 88 def dataset_fn(filenames, num_epochs, batch_size=None): function
|