Home
last modified time | relevance | path

Searched refs:with_options (Results 1 – 25 of 58) sorted by relevance

123

/external/tensorflow/tensorflow/python/data/kernel_tests/
Doptions_test.py37 ds = dataset_ops.Dataset.range(0).with_options(options).cache()
43 ds = dataset_ops.Dataset.range(0).with_options(options).with_options(
52 ds = dataset_ops.Dataset.range(0).with_options(options1).with_options(
65 dataset_ops.Dataset.range(0).with_options(options1).with_options(options2)
73 (dataset_ops.Dataset.range(0).with_options(options1),
74 dataset_ops.Dataset.range(0).with_options(options2)))
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Doptimize_dataset_test.py111 dataset = dataset.with_options(options)
121 dataset = dataset.with_options(options)
136 dataset = dataset.with_options(options)
159 dataset = dataset.with_options(options)
178 dataset = dataset.with_options(options)
191 dataset = dataset.with_options(options)
208 dataset = dataset.with_options(options)
218 dataset = dataset.with_options(options)
238 optimized_dataset = unoptimized_dataset.with_options(options)
Dmap_vectorization_test.py356 dataset = dataset.with_options(options)
367 optimized = optimized.with_options(options)
417 base_dataset = base_dataset.with_options(options)
428 optimized = optimized.with_options(options)
464 base_dataset = base_dataset.with_options(options)
479 optimized = optimized.with_options(options)
537 unoptimized = unoptimized.with_options(options)
541 optimized = unoptimized.with_options(options)
556 unoptimized = unoptimized.with_options(options)
560 optimized = unoptimized.with_options(options)
[all …]
Dassert_next_dataset_test.py36 dataset = dataset.with_options(options)
44 dataset = dataset.with_options(options)
58 dataset = dataset.with_options(options)
Dmap_parallelization_test.py62 dataset = dataset.with_options(options)
77 dataset = dataset.with_options(options)
92 dataset = dataset.with_options(options)
Dhoist_random_uniform_test.py90 dataset = dataset.with_options(options)
107 dataset = dataset.with_options(options)
Dmap_and_filter_fusion_test.py88 dataset = dataset.with_options(options)
107 dataset = dataset.with_options(options)
Dmap_and_batch_fusion_test.py37 dataset = dataset.with_options(options)
Dmake_numa_aware_test.py38 dataset = dataset.with_options(options)
Dmodel_dataset_test.py40 dataset = dataset.with_options(options)
Dshuffle_and_repeat_fusion_test.py37 dataset = dataset.with_options(options)
Dnoop_elimination_test.py47 dataset = dataset.with_options(options)
Dlatency_all_edges_test.py41 dataset = dataset.with_options(options)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dmap_and_batch_test.py79 dataset = dataset.with_options(options)
145 dataset = dataset.with_options(options)
170 dataset = dataset.with_options(options)
187 dataset = dataset.with_options(options)
222 dataset = dataset.with_options(options)
260 dataset = dataset.with_options(options)
286 dataset = dataset.with_options(options)
310 dataset = dataset.with_options(options)
338 dataset = dataset.with_options(options)
372 dataset = dataset.with_options(options)
[all …]
/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
Doptimize_benchmark.py52 dataset = dataset.with_options(options)
94 dataset = dataset.with_options(options)
134 dataset = dataset.with_options(options)
Dautotune_benchmark.py50 dataset = dataset.with_options(options)
89 dataset = dataset.with_options(options)
127 dataset = dataset.with_options(options)
187 dataset = dataset.with_options(options)
Dmap_and_batch_benchmark.py59 dataset = dataset.with_options(options)
139 return dataset.with_options(options)
170 fused_dataset = chained_dataset.with_options(options)
Dchoose_fastest_benchmark.py39 dataset = dataset.with_options(options)
54 dataset = dataset.with_options(options)
Dunbatch_benchmark.py47 dataset = dataset.with_options(options)
84 dataset = dataset.with_options(options)
Dmap_vectorization_benchmark.py123 unoptimized_dataset = unoptimized_dataset.with_options(options)
129 optimized_dataset = unoptimized_dataset.with_options(options)
Drejection_resample_benchmark.py44 dataset = dataset.with_options(options)
Dparallel_interleave_benchmark.py61 dataset = dataset_fn().with_options(options)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dnuma_map_and_batch_dataset_serialization_test.py55 return ds.with_options(options)
85 return ds.with_options(options)
/external/tensorflow/tensorflow/python/data/benchmarks/
Drange_benchmark.py33 dataset = dataset.with_options(options)
Dbenchmark_base.py52 dataset = dataset.with_options(options)

123