Home
last modified time | relevance | path

Searched refs:run_core_tests (Results 1 – 25 of 39) sorted by relevance

12

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dsequence_dataset_serialization_test.py37 self.run_core_tests(lambda: self._build_skip_dataset(count),
43 self.run_core_tests(lambda: self._build_skip_dataset(20), None, 0)
45 self.run_core_tests(lambda: self._build_skip_dataset(10), None, 0)
46 self.run_core_tests(lambda: self._build_skip_dataset(-1), None, 0)
48 self.run_core_tests(lambda: self._build_skip_dataset(0), None, 10)
53 self.run_core_tests(lambda: self._build_skip_dataset([1, 2]), None, 0)
65 self.run_core_tests(
73 self.run_core_tests(lambda: self._build_take_dataset(20), None, 10)
75 self.run_core_tests(lambda: self._build_take_dataset(10), None, 10)
77 self.run_core_tests(lambda: self._build_take_dataset(-1), None, 10)
[all …]
Dstats_dataset_serialization_test.py45 self.run_core_tests(
53 self.run_core_tests(
72 self.run_core_tests(
81 self.run_core_tests(
86 self.run_core_tests(lambda: self._build_dataset_multiple_tags(num_outputs),
91 self.run_core_tests(
103 self.run_core_tests(self._build_dataset_stats_aggregator, None, 10)
Dmap_and_batch_dataset_serialization_test.py55 self.run_core_tests(lambda: build_ds(10), lambda: build_ds(15),
57 self.run_core_tests(lambda: build_ds(10, True), lambda: build_ds(15, True),
82 self.run_core_tests(lambda: build_ds(10), lambda: build_ds(15),
84 self.run_core_tests(lambda: build_ds(10, True), lambda: build_ds(15, True),
98 self.run_core_tests(build_dataset, None, 2)
Dtf_record_dataset_serialization_test.py70 self.run_core_tests(
75 self.run_core_tests(
83 self.run_core_tests(lambda: self._build_iterator_graph(num_epochs),
90 self.run_core_tests(
93 self.run_core_tests(
Dmap_dataset_serialization_test.py56 self.run_core_tests(
90 self.run_core_tests(_build_ds, None, 10)
103 self.run_core_tests(_build_ds, None, num_outputs)
122 self.run_core_tests(_build_ds, None, num_outputs)
136 self.run_core_tests(lambda: _build_ds(num_outputs),
Dbatch_dataset_serialization_test.py45 self.run_core_tests(
60 self.run_core_tests(lambda: self._build_dataset_dense_to_sparse(components),
72 self.run_core_tests(self._build_dataset_sparse,
79 self.run_core_tests(self._build_dataset_nested_sparse, None, 1)
Dnuma_map_and_batch_dataset_serialization_test.py57 self.run_core_tests(lambda: build_ds(10), lambda: build_ds(15),
59 self.run_core_tests(lambda: build_ds(10, True), lambda: build_ds(15, True),
87 self.run_core_tests(lambda: build_ds(10), lambda: build_ds(15),
89 self.run_core_tests(lambda: build_ds(10, True), lambda: build_ds(15, True),
Ddataset_constructor_serialization_test.py41 self.run_core_tests(lambda: self._build_tensor_dataset(arr),
64 self.run_core_tests(lambda: self._build_tensor_slices_dataset(components),
66 self.run_core_tests(
87 self.run_core_tests(
Dchoose_fastest_branch_dataset_serialization_test.py49 …self.run_core_tests(lambda: build_ds(size), None, size // 10) # pylint: disable=cell-var-from-loop
67 self.run_core_tests(build_ds, None, 10)
85 self.run_core_tests(build_ds, None, 10)
100 self.run_core_tests(build_ds, None, 1000)
Dparallel_interleave_dataset_serialization_test.py49 self.run_core_tests(
56 self.run_core_tests(lambda: self._build_ds(cycle_length, block_length),
61 self.run_core_tests(lambda: self._build_ds(cycle_length, block_length),
97 self.run_core_tests(_build_dataset, None, 20)
Dflat_map_dataset_serialization_test.py46 self.run_core_tests(lambda: build_ds(0), lambda: build_ds(10), 25)
61 self.run_core_tests(build_ds, None, 500)
77 self.run_core_tests(build_ds, None, 100)
118 self.run_core_tests(_build_ds, None, 20)
Dfilter_dataset_serialization_test.py37 self.run_core_tests(lambda: self._build_filter_range_graph(div),
49 self.run_core_tests(self._build_filter_dict_graph, None, num_outputs)
65 self.run_core_tests(self._build_sparse_filter, None, num_outputs)
Dparallel_map_dataset_serialization_test.py66 self.run_core_tests(
102 self.run_core_tests(_build_ds, None, 10)
116 self.run_core_tests(_build_ds, None, num_outputs)
136 self.run_core_tests(_build_ds, None, num_outputs)
Dzip_dataset_serialization_test.py46 self.run_core_tests(lambda: self._build_dataset(arr), None, num_outputs)
49 self.run_core_tests(lambda: self._build_dataset(diff_size_arr),
Doptimize_dataset_serialization_test.py35 self.run_core_tests(lambda: build_dataset(200, 10), None, 20)
49 self.run_core_tests(build_dataset, None, 20)
Dpadded_batch_dataset_serialization_test.py41 self.run_core_tests(lambda: build_dataset(seq_lens1),
61 self.run_core_tests(lambda: build_dataset(seq_lens1),
Dinterleave_dataset_serialization_test.py57 self.run_core_tests(
79 self.run_core_tests(_build_dataset, None, 20)
Dprefetch_dataset_serialization_test.py34 self.run_core_tests(lambda: self.build_dataset(10),
Drebatch_dataset_serialization_test.py37 self.run_core_tests(lambda: build_dataset(200, 10), None, 20)
Dunique_dataset_serialization_test.py35 self.run_core_tests(lambda: build_dataset(200, 100),
Dshuffle_and_repeat_dataset_serialization_test.py34 self.run_core_tests(lambda: self._build_ds(10), lambda: self._build_ds(20),
Dchoose_fastest_dataset_serialization_test.py41 self.run_core_tests(build_ds, None, num_outputs // 2)
Dsample_from_datasets_serialization_test.py40 self.run_core_tests(
Dscan_dataset_serialization_test.py35 self.run_core_tests(lambda: self._build_dataset(num_output),
Dshard_dataset_serialization_test.py36 self.run_core_tests(lambda: self._build_dataset(elems, num_shards, index1),

12