Home
last modified time | relevance | path

Searched refs:Dataset (Results 1 – 25 of 443) sorted by relevance

12345678910>>...18

/external/tensorflow/tensorflow/python/data/kernel_tests/
Dcardinality_test.py33 ("Map1", lambda: dataset_ops.Dataset.range(5).map(lambda x: x),
35 ("Map2", lambda: dataset_ops.Dataset.range(5).map(
39 ("Map1", lambda: dataset_ops.Dataset.range(5).map(lambda x: x), 5),
40 ("Map2", lambda: dataset_ops.Dataset.range(5).map(
45 lambda: dataset_ops.Dataset.range(5).batch(2, drop_remainder=True), 2),
47 lambda: dataset_ops.Dataset.range(5).batch(2, drop_remainder=False), 3),
49 lambda: dataset_ops.Dataset.range(5).filter(lambda _: True).batch(2),
51 ("Batch4", lambda: dataset_ops.Dataset.range(5).repeat().batch(2),
53 ("Cache1", lambda: dataset_ops.Dataset.range(5).cache(), 5),
54 ("Cache2", lambda: dataset_ops.Dataset.range(5).cache("foo"), 5),
[all …]
Ddataset_test.py54 dataset = dataset_ops.Dataset.range(10)
60 dataset = dataset_ops.Dataset.range(10).map(
72 original_dataset = dataset_ops.Dataset.range(5).map(lambda x: x * 2)
85 original_dataset = dataset_ops.Dataset.range(5).flat_map(
86 lambda x: dataset_ops.Dataset.range(5).map(lambda x: x * 2))
107 dataset = dataset_ops.Dataset.from_generator(gen, dtypes.int32)
112 dataset = dataset_ops.Dataset.from_tensors([42])
117 dataset = dataset_ops.Dataset.range(10)
133 dataset_fn = dataset_ops.Dataset.from_sparse_tensor_slices(
141 input_dataset = dataset_ops.Dataset.range(0)
[all …]
Dshard_test.py33 dataset = dataset_ops.Dataset.range(10).shard(5, 2)
38 dataset_a = dataset_ops.Dataset.range(10)
39 dataset_b = dataset_ops.Dataset.range(10, 0, -1)
40 dataset = dataset_ops.Dataset.zip((dataset_a, dataset_b)).shard(5, 2)
45 dataset = dataset_ops.Dataset.range(10).shard(5, 0)
51 dataset = dataset_ops.Dataset.range(10).shard(5, 7)
57 dataset = dataset_ops.Dataset.range(10).shard(5, -3)
63 dataset = dataset_ops.Dataset.range(10).shard(-3, 1)
69 dataset = dataset_ops.Dataset.range(10).shard(0, 1)
74 dataset = dataset_ops.Dataset.range(1).shard(5, 2)
[all …]
Dflat_map_test.py48 dataset = dataset_ops.Dataset.from_tensor_slices(components).flat_map(
49 lambda x: dataset_ops.Dataset.from_tensors([x]).repeat(x))
59 dataset = dataset_ops.Dataset.from_tensor_slices(components).flat_map(
60 lambda x: dataset_ops.Dataset.from_tensor_slices(x).flat_map(
61 lambda y: dataset_ops.Dataset.from_tensors(y).repeat(y))
75 dataset_ops.Dataset.from_tensor_slices(components).flat_map(
76 lambda x: dataset_ops.Dataset.from_tensor_slices(x).flat_map(
77 lambda y: dataset_ops.Dataset.from_tensors(y).repeat(y))),
104 dataset = dataset_ops.Dataset.range(10).map(
106 lambda d: dataset_ops.Dataset.from_tensors(
[all …]
Dpadded_batch_test.py52 dataset = dataset_ops.Dataset.from_tensor_slices(seq_lens).map(
90 dataset_ops.Dataset.from_tensor_slices(
99 dataset_ops.Dataset.from_tensor_slices(
111 dataset_ops.Dataset.from_tensor_slices(
124 dataset_ops.Dataset.from_tensor_slices(
147 dataset_ops.Dataset.from_tensor_slices(random_seq_lens).map(fill_tuple)
185 dataset = dataset_ops.Dataset.from_generator(
198 input_dataset = dataset_ops.Dataset.from_tensors(
233 _ = dataset_ops.Dataset.from_tensors(st).repeat(10).padded_batch(10)
242 _ = dataset_ops.Dataset.from_tensors(rt).repeat(10).padded_batch(10)
[all …]
Dfrom_generator_test.py44 dataset = dataset_ops.Dataset.from_generator(
136 return (dataset_ops.Dataset.from_generator(
141 dataset = dataset_ops.Dataset.range(num_outer_repeats).interleave(
190 return dataset_ops.Dataset.from_generator(
193 dataset = dataset_ops.Dataset.range(num_parallel_iterators).interleave(
211 dataset = dataset_ops.Dataset.from_generator(
229 dataset = dataset_ops.Dataset.from_generator(
242 dataset = dataset_ops.Dataset.from_generator(
260 dataset = dataset_ops.Dataset.from_generator(
281 dataset = dataset_ops.Dataset.from_generator(
[all …]
Das_numpy_iterator_test.py38 ds = dataset_ops.Dataset.range(3)
43 ds = dataset_ops.Dataset.from_tensors([1, 2, 3])
52 ds = dataset_ops.Dataset.from_tensor_slices({
69 ds = dataset_ops.Dataset.range(10)
74 ds = dataset_ops.Dataset.from_tensors(element)
87 ds = dataset_ops.Dataset.from_tensor_slices(rt)
93 self._testInvalidElement(dataset_ops.Dataset.range(3))
97 tuple_elem = (constant_op.constant([1, 2, 3]), dataset_ops.Dataset.range(3))
Dmemory_cleanup_test.py77 dataset = dataset_ops.Dataset.range(10)
95 dataset = dataset_ops.Dataset.range(10)
131 return dataset_ops.Dataset.range(0, 100).filter(fn)
143 return dataset_ops.Dataset.range(0, 100).filter_with_legacy_function(fn)
153 return dataset_ops.Dataset.from_tensors(x * x)
155 return dataset_ops.Dataset.range(0, 100).flat_map(fn)
167 return dataset_ops.Dataset.from_generator(fn, output_types=dtypes.float32)
181 return dataset_ops.Dataset.range(0, 100).map(
196 return dataset_ops.Dataset.range(0, 100).map_with_legacy_function(
209 return dataset_ops.Dataset.from_tensors(x * x)
[all …]
Dconcatenate_test.py45 input_dataset = dataset_ops.Dataset.from_tensor_slices(input_components)
46 dataset_to_concatenate = dataset_ops.Dataset.from_tensor_slices(
77 input_dataset = dataset_ops.Dataset.from_tensor_slices(input_components)
78 dataset_to_concatenate = dataset_ops.Dataset.from_tensor_slices(
109 input_dataset = dataset_ops.Dataset.from_tensor_slices(input_components)
110 dataset_to_concatenate = dataset_ops.Dataset.from_tensor_slices(
127 input_dataset = dataset_ops.Dataset.from_tensor_slices(input_components)
128 dataset_to_concatenate = dataset_ops.Dataset.from_tensor_slices(
143 input_dataset = dataset_ops.Dataset.from_tensor_slices(input_components)
144 dataset_to_concatenate = dataset_ops.Dataset.from_tensor_slices(
Dunbatch_test.py42 dataset = dataset_ops.Dataset.from_tensors([0, 1, 2, 3]).unbatch()
48 data = dataset_ops.Dataset.from_tensor_slices(data)
59 data = dataset_ops.Dataset.from_tensors(
60 [dataset_ops.Dataset.range(10) for _ in range(10)])
67 data = dataset_ops.Dataset.from_tensor_slices(data)
84 data = dataset_ops.Dataset.from_tensors(st)
101 data = dataset_ops.Dataset.from_tensors((list(range(10)), st, rt))
115 data = dataset_ops.Dataset.from_tensors(rt)
130 data = dataset_ops.Dataset.from_tensor_slices(data)
143 data = dataset_ops.Dataset.from_tensor_slices(data)
[all …]
Dreduce_test.py47 ds = dataset_ops.Dataset.range(1, i + 1)
59 ds = dataset_ops.Dataset.range(1, i + 1)
60 ds = dataset_ops.Dataset.zip((ds, ds))
72 ds = dataset_ops.Dataset.range(1, i + 1)
88 ds = dataset_ops.Dataset.range(1, i + 1)
107 ds = dataset_ops.Dataset.from_tensors(make_sparse_fn(i+1))
130 ds = dataset_ops.Dataset.range(1, i + 1).map(map_fn)
145 return dataset_ops.Dataset.range(10).map(increment_fn)
164 return dataset_ops.Dataset.range(10)
184 return dataset_ops.Dataset.range(1)
[all …]
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Ddirected_interleave_dataset_test.py39 selector_dataset = dataset_ops.Dataset.range(10).repeat(100)
41 dataset_ops.Dataset.from_tensors(i).repeat(100) for i in range(10)
67 dataset_ops.Dataset.from_tensors(i).repeat(None)
97 probs_ds = dataset_ops.Dataset.from_tensors(probs).repeat()
103 ds1 = dataset_ops.Dataset.from_tensors([1.0]).repeat()
104 ds2 = dataset_ops.Dataset.from_tensors([2.0]).repeat()
111 datasets = [dataset_ops.Dataset.from_tensors(w).repeat() for w in words]
113 choice_dataset = dataset_ops.Dataset.from_tensor_slices(choice_array)
126 [dataset_ops.Dataset.range(10),
127 dataset_ops.Dataset.range(20)],
[all …]
Drebatch_dataset_test.py168 dataset = dataset_ops.Dataset.range(8).batch(4, drop_remainder=True)
178 dataset = dataset_ops.Dataset.range(8).batch(4, drop_remainder=True)
187 dataset = dataset_ops.Dataset.range(8).batch(4, drop_remainder=False)
196 dataset = dataset_ops.Dataset.range(8).batch(4, drop_remainder=False)
205 dataset = dataset_ops.Dataset.range(10).batch(5, drop_remainder=True)
214 dataset = dataset_ops.Dataset.range(10).batch(5, drop_remainder=True)
227 dataset = dataset_ops.Dataset.range(8).batch(4, drop_remainder=True)
240 dataset = dataset_ops.Dataset.range(5).batch(4, drop_remainder=False)
252 dataset = dataset_ops.Dataset.range(5).batch(4, drop_remainder=False)
265 dataset = dataset_ops.Dataset.range(12).batch(
[all …]
Ddata_service_ops_test.py75 ds = dataset_ops.Dataset.from_tensors(element)
83 ds = dataset_ops.Dataset.from_tensor_slices([1, 5, 3, 2, 8])
97 ds = dataset_ops.Dataset.range(num_elements)
220 return dataset_ops.Dataset.range(num_elements).shuffle(num_elements)
294 ds = dataset_ops.Dataset.range(10000000)
306 ds = dataset_ops.Dataset.from_tensor_slices(consumers)
338 ds = dataset_ops.Dataset.range(num_elements, output_type=dtypes.int32)
350 lambda _, x: dataset_ops.Dataset.from_tensors(x),
368 host_dataset = dataset_ops.Dataset.from_tensor_slices(
378 ds = dataset_ops.Dataset.from_tensor_slices(host_datasets)
[all …]
Dsnapshot_test.py109 dataset = dataset_ops.Dataset.from_tensors([1, 2, 3])
223 dataset = dataset_ops.Dataset.range(1000)
233 dataset = dataset_ops.Dataset.range(1000)
242 dataset = dataset_ops.Dataset.range(0)
251 dataset2 = dataset_ops.Dataset.range(0)
257 dataset = dataset_ops.Dataset.range(1000)
268 dataset1 = dataset_ops.Dataset.range(1000)
272 dataset2 = dataset_ops.Dataset.range(2000)
284 dataset1 = dataset_ops.Dataset.range(1000)
287 dataset2 = dataset_ops.Dataset.range(1000)
[all …]
Dio_test.py50 dataset = dataset_ops.Dataset.range(42)
58 dataset = dataset_ops.Dataset.range(42)
65 dataset = dataset_ops.Dataset.range(42)
75 dataset = dataset_ops.Dataset.range(42)
91 dataset = dataset_ops.Dataset.range(42)
104 range_dataset = dataset_ops.Dataset.range(42)
105 dict_dataset = dataset_ops.Dataset.from_tensor_slices({"a": [1, 2],
107 tuple_dataset = dataset_ops.Dataset.from_tensor_slices(([1, 2], [3, 4]))
108 dataset = dataset_ops.Dataset.zip((range_dataset, dict_dataset,
Dgroup_by_window_test.py48 return dataset_ops.Dataset.zip(
49 (dataset_ops.Dataset.from_tensors(bucket),
61 input_dataset = dataset_ops.Dataset.from_tensor_slices(
91 input_dataset = dataset_ops.Dataset.from_tensor_slices(
148 return dataset_ops.Dataset.zip(
149 (dataset_ops.Dataset.from_tensors(bucket),
157 input_dataset = dataset_ops.Dataset.from_tensor_slices(math_ops.range(
191 dataset = dataset_ops.Dataset.from_tensor_slices(components).apply(
212 dataset = dataset_ops.Dataset.from_tensor_slices(
234 dataset = dataset_ops.Dataset.from_tensor_slices(components).repeat(
[all …]
Dgroup_by_reducer_test.py46 dataset = dataset_ops.Dataset.range(2 * i).apply(
65 dataset = dataset_ops.Dataset.range(2 * i).apply(
81 dataset = dataset_ops.Dataset.zip(
82 (dataset_ops.Dataset.from_tensor_slices(components),
83 dataset_ops.Dataset.range(2 * i))).apply(
103 dataset = dataset_ops.Dataset.range(2 * i).map(_sparse).apply(
126 dataset = dataset_ops.Dataset.from_tensors(np.int64(0)).repeat(i).apply(
145 dataset = dataset_ops.Dataset.range(10)
160 dataset = dataset_ops.Dataset.range(10)
174 dataset = dataset_ops.Dataset.range(10)
[all …]
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dflat_map_dataset_serialization_test.py48 return dataset_ops.Dataset.range(x, x + 5)
50 return dataset_ops.Dataset.range(start, start + 5 * 5, 5).flat_map(map_fn)
64 return dataset_ops.Dataset.range(100).map(map_fn)
66 return dataset_ops.Dataset.range(5).flat_map(flat_map_fn)
81 return dataset_ops.Dataset.from_tensor_slices([defun_fn(x)])
83 return dataset_ops.Dataset.range(100).flat_map(map_fn)
93 return dataset_ops.Dataset.range(5).flat_map(
94 lambda _: dataset_ops.Dataset.from_tensor_slices([test_var]))
109 return dataset_ops.Dataset.range(100).map(map_fn)
111 return dataset_ops.Dataset.range(5).flat_map(flat_map_fn)
[all …]
/external/tensorflow/tensorflow/python/data/util/
Dtraverse_test.py46 ds = dataset_ops.Dataset.range(10)
52 ds = dataset_ops.Dataset.range(10).map(math_ops.square)
60 ds1 = dataset_ops.Dataset.range(10)
61 ds2 = dataset_ops.Dataset.range(10)
70 ds1 = dataset_ops.Dataset.range(10)
71 ds2 = dataset_ops.Dataset.range(10)
72 ds = dataset_ops.Dataset.zip((ds1, ds2))
80 ds = dataset_ops.Dataset.range(10)
89 ds1 = dataset_ops.Dataset.range(10).repeat(10)
98 ds2 = dataset_ops.Dataset.range(20).prefetch(1)
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_utils_v1_test.py142 ('Batch', lambda: dataset_ops.Dataset.range(5).batch(2)),
143 ('Cache', lambda: dataset_ops.Dataset.range(5).cache()),
144 ('Concatenate', lambda: dataset_ops.Dataset.range(5).concatenate(
145 dataset_ops.Dataset.range(5))),
146 ('FlatMap', lambda: dataset_ops.Dataset.range(5).flat_map(
147 lambda _: dataset_ops.Dataset.from_tensors(0))),
148 ('FlatMap_Shuffle', lambda: dataset_ops.Dataset.range(5).flat_map(
149 lambda _: dataset_ops.Dataset.from_tensors(0).shuffle(1)), True),
150 ('Filter', lambda: dataset_ops.Dataset.range(5).filter(lambda _: True)),
153 ('FromTensors', lambda: dataset_ops.Dataset.from_tensors(0)),
[all …]
/external/tensorflow/tensorflow/core/data/
Dstandalone.cc45 Status Dataset::FromGraph(Params params, const GraphDef& graph_def, in FromGraph()
46 std::unique_ptr<Dataset>* result) { in FromGraph()
96 WrapUnique(new Dataset(dataset, device_mgr.release(), pflr.release(), in FromGraph()
101 Status Dataset::MakeIterator(std::unique_ptr<SplitProvider> split_provider, in MakeIterator()
134 Status Dataset::MakeIterator(std::unique_ptr<Iterator>* result) { in MakeIterator()
138 Status Dataset::MakeSplitProvider(std::unique_ptr<SplitProvider>* result) { in MakeSplitProvider()
142 const DatasetBase* Dataset::Get() const { return dataset_; } in Get()
144 Dataset::Dataset(DatasetBase* dataset, DeviceMgr* device_mgr, in Dataset() function in tensorflow::data::standalone::Dataset
157 Dataset::~Dataset() { dataset_->Unref(); } in ~Dataset()
Dstandalone.h65 class Dataset; variable
76 friend class Dataset;
86 class Dataset {
95 std::unique_ptr<Dataset>* result);
97 ~Dataset();
111 Dataset(DatasetBase* dataset, DeviceMgr* device_mgr,
/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
Dautotune_benchmark.py61 dataset = dataset_ops.Dataset.from_tensors(
83 dataset = dataset_ops.Dataset.from_tensors(
105 dataset = dataset_ops.Dataset.from_tensors(
127 dataset = dataset_ops.Dataset.from_tensors(
130 dataset = dataset_ops.Dataset.range(1).repeat().interleave(
155 dataset_a = dataset_ops.Dataset.from_tensors(a).repeat()
156 dataset_b = dataset_ops.Dataset.from_tensors(b).repeat()
157 dataset_c = dataset_ops.Dataset.from_tensors(c).repeat()
168 dataset = dataset_ops.Dataset.range(1).repeat().interleave(
173 dataset = dataset_ops.Dataset.zip((dataset, dataset_b))
[all …]
/external/tensorflow/tensorflow/python/keras/preprocessing/
Dtimeseries.py198 positions_ds = dataset_ops.Dataset.from_tensors(start_positions).repeat()
201 indices = dataset_ops.Dataset.zip(
202 (dataset_ops.Dataset.range(len(start_positions)), positions_ds)).map(
211 indices = dataset_ops.Dataset.zip(
212 (dataset_ops.Dataset.range(len(start_positions)), positions_ds)).map(
217 dataset = dataset_ops.Dataset.zip((dataset, target_ds))
226 dataset = dataset_ops.Dataset.from_tensors(array[start_index : end_index])
227 dataset = dataset_ops.Dataset.zip((dataset.repeat(), indices_ds)).map(

12345678910>>...18