Home
last modified time | relevance | path

Searched refs:CreateTensors (Results 1 – 25 of 45) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/data/
Dwindow_dataset_op_test.cc247 {CreateTensors<int64>(TensorShape{}, {{0}, {1}}), in GetNextTestCases()
248 CreateTensors<int64>(TensorShape{}, {{2}, {3}}), in GetNextTestCases()
249 CreateTensors<int64>(TensorShape{}, {{4}, {5}}), in GetNextTestCases()
250 CreateTensors<int64>(TensorShape{}, {{6}})}}, in GetNextTestCases()
253 {CreateTensors<int64>(TensorShape{}, {{0}, {2}}), in GetNextTestCases()
254 CreateTensors<int64>(TensorShape{}, {{2}, {4}}), in GetNextTestCases()
255 CreateTensors<int64>(TensorShape{}, {{4}, {6}})}}, in GetNextTestCases()
258 {CreateTensors<int64>(TensorShape({}), in GetNextTestCases()
260 CreateTensors<int64>(TensorShape({}), {{3}, {4}, {5}, {6}}), in GetNextTestCases()
261 CreateTensors<int64>(TensorShape({}), {{6}})}}, in GetNextTestCases()
[all …]
Dpadded_batch_dataset_op_test.cc126 /*components=*/CreateTensors<int64>(TensorShape{3, 2}, in PaddedBatchDatasetParams2()
130 /*components=*/CreateTensors<int64>(TensorShape{4, 1}, {{6, 7, 8, 9}}), in PaddedBatchDatasetParams2()
154 /*components=*/CreateTensors<int64>(TensorShape{3, 2}, in PaddedBatchDatasetParams3()
158 /*components=*/CreateTensors<int64>(TensorShape{4, 1}, {{6, 7, 8, 9}}), in PaddedBatchDatasetParams3()
184 /*components=*/CreateTensors<int64>(TensorShape{3, 2}, in PaddedBatchDatasetParams4()
188 /*components=*/CreateTensors<int64>(TensorShape{3, 1}, {{6, 7, 8}}), in PaddedBatchDatasetParams4()
212 /*components=*/CreateTensors<int64>(TensorShape{3, 2}, in PaddedBatchDatasetParams5()
216 /*components=*/CreateTensors<int64>(TensorShape{4, 1}, {{6, 7, 8, 9}}), in PaddedBatchDatasetParams5()
240 /*components=*/CreateTensors<int64>(TensorShape{3, 2}, in PaddedBatchDatasetParams6()
244 /*components=*/CreateTensors<int64>(TensorShape{4, 1}, {{6, 7, 8, 9}}), in PaddedBatchDatasetParams6()
[all …]
Dinterleave_dataset_op_test.cc310 CreateTensors<int64>(TensorShape({1}), in GetNextTestCases()
313 /*expected_outputs=*/CreateTensors<int64>( in GetNextTestCases()
316 /*expected_outputs=*/CreateTensors<int64>( in GetNextTestCases()
319 /*expected_outputs=*/CreateTensors<int64>( in GetNextTestCases()
322 /*expected_outputs=*/CreateTensors<tstring>( in GetNextTestCases()
326 /*expected_outputs=*/CreateTensors<tstring>( in GetNextTestCases()
330 /*expected_outputs=*/CreateTensors<tstring>( in GetNextTestCases()
342 CreateTensors<int64>(TensorShape({1}), {{0}})}, in ITERATOR_GET_NEXT_TEST_P()
346 CreateTensors<int64>(TensorShape({1}), {{5}})}, in ITERATOR_GET_NEXT_TEST_P()
352 CreateTensors<int64>(TensorShape({1}), {{5}})}, in ITERATOR_GET_NEXT_TEST_P()
[all …]
Dshuffle_dataset_op_test.cc259 CreateTensors<int64>(TensorShape({}), in GetNextTestCases()
262 CreateTensors<int64>( in GetNextTestCases()
267 CreateTensors<int64>(TensorShape({}), in GetNextTestCases()
270 CreateTensors<int64>( in GetNextTestCases()
275 CreateTensors<int64>(TensorShape({}), in GetNextTestCases()
278 CreateTensors<int64>( in GetNextTestCases()
283 CreateTensors<int64>(TensorShape({}), in GetNextTestCases()
286 CreateTensors<int64>( in GetNextTestCases()
291 CreateTensors<int64>(TensorShape({}), in GetNextTestCases()
294 CreateTensors<int64>( in GetNextTestCases()
[all …]
Drange_dataset_op_test.cc50 CreateTensors<int64>(TensorShape({}), {{0}, {3}, {6}, {9}})}, in GetNextTestCases()
53 CreateTensors<int64>(TensorShape({}), {{10}, {7}, {4}, {1}})}}; in GetNextTestCases()
128 CreateTensors<int64>(TensorShape({}), {{0}, {3}, {6}, {9}})}, in IteratorSaveAndRestoreTestCases()
132 CreateTensors<int64>(TensorShape({}), {{10}, {7}, {4}, {1}})}}; in IteratorSaveAndRestoreTestCases()
149 params, CreateTensors<int64>(TensorShape({}), {{0}, {3}, {6}, {9}}))); in TEST_F()
152 CreateTensors<int64>(TensorShape({}), {{3}, {9}}))); in TEST_F()
160 params, CreateTensors<int64>(TensorShape({}), {{10}, {7}, {4}, {1}}))); in TEST_F()
163 CreateTensors<int64>(TensorShape({}), {{10}, {4}}))); in TEST_F()
171 params, CreateTensors<int64>(TensorShape({}), {}))); in TEST_F()
174 CreateTensors<int64>(TensorShape({}), {}))); in TEST_F()
Dshard_dataset_op_test.cc41 return CreateTensors<int64>(TensorShape({}), {{num_shards_}, {index_}}); in GetInputTensors()
208 /*expected_outputs=*/CreateTensors<int64>(TensorShape{}, {{2}, {7}})}, in GetNextTestCases()
210 /*expected_outputs=*/CreateTensors<int64>(TensorShape{}, {{0}, {5}})}, in GetNextTestCases()
214 /*expected_outputs=*/CreateTensors<int64>(TensorShape{}, {{5}})}, in GetNextTestCases()
216 /*expected_outputs=*/CreateTensors<int64>(TensorShape{}, {{4}, {9}})}, in GetNextTestCases()
218 /*expected_outputs=*/CreateTensors<int64>(TensorShape{}, {{3}, {7}})}, in GetNextTestCases()
220 /*expected_outputs=*/CreateTensors<int64>(TensorShape{}, {{5}})}}; in GetNextTestCases()
295 /*expected_outputs=*/CreateTensors<int64>(TensorShape{}, {{2}, {7}})}, in IteratorSaveAndRestoreTestCases()
298 /*expected_outputs=*/CreateTensors<int64>(TensorShape{}, {{0}, {5}})}, in IteratorSaveAndRestoreTestCases()
304 /*expected_outputs=*/CreateTensors<int64>(TensorShape{}, {{5}})}, in IteratorSaveAndRestoreTestCases()
[all …]
Dtf_record_dataset_op_test.cc144 CreateTensors<tstring>( in GetNextTestCases()
147 CreateTensors<tstring>( in GetNextTestCases()
150 CreateTensors<tstring>( in GetNextTestCases()
161 CreateTensors<tstring>(TensorShape({}), {{"333"}})}, in ITERATOR_GET_NEXT_TEST_P()
165 CreateTensors<tstring>(TensorShape({}), {{"bb"}})}, in ITERATOR_GET_NEXT_TEST_P()
172 CreateTensors<tstring>(TensorShape({}), {{"333"}})}, in ITERATOR_GET_NEXT_TEST_P()
176 CreateTensors<tstring>(TensorShape({}), {{"bb"}})}, in ITERATOR_GET_NEXT_TEST_P()
183 CreateTensors<tstring>(TensorShape({}), {{"333"}})}, in ITERATOR_GET_NEXT_TEST_P()
187 CreateTensors<tstring>(TensorShape({}), {{"bb"}})}, in ITERATOR_GET_NEXT_TEST_P()
251 CreateTensors<tstring>( in IteratorSaveAndRestoreTestCases()
[all …]
Dparallel_interleave_dataset_op_test.cc536 CreateTensors<int64>(TensorShape{1}, in GetNextTestCases()
541 CreateTensors<int64>(TensorShape{1}, in GetNextTestCases()
546 CreateTensors<int64>(TensorShape{1}, in GetNextTestCases()
551 CreateTensors<int64>(TensorShape{1}, in GetNextTestCases()
556 CreateTensors<tstring>( in GetNextTestCases()
563 CreateTensors<tstring>( in GetNextTestCases()
570 CreateTensors<tstring>( in GetNextTestCases()
577 CreateTensors<tstring>( in GetNextTestCases()
584 CreateTensors<tstring>( in GetNextTestCases()
591 CreateTensors<tstring>( in GetNextTestCases()
[all …]
Dprefetch_dataset_op_test.cc201 CreateTensors<int64>( in GetNextTestCases()
205 CreateTensors<int64>( in GetNextTestCases()
210 CreateTensors<int64>( in GetNextTestCases()
215 CreateTensors<int64>( in GetNextTestCases()
220 CreateTensors<int64>( in GetNextTestCases()
225 CreateTensors<int64>( in GetNextTestCases()
299 CreateTensors<int64>( in IteratorSaveAndRestoreTestCases()
304 CreateTensors<int64>( in IteratorSaveAndRestoreTestCases()
310 CreateTensors<int64>( in IteratorSaveAndRestoreTestCases()
316 CreateTensors<int64>( in IteratorSaveAndRestoreTestCases()
[all …]
Dparallel_batch_dataset_op_test.cc253 CreateTensors<int64>(TensorShape({4}), in GetNextTestCases()
257 CreateTensors<int64>(TensorShape({4}), in GetNextTestCases()
267 CreateTensors<int64>(TensorShape({3}), in GetNextTestCases()
273 CreateTensors<int64>(TensorShape({10}), in GetNextTestCases()
279 CreateTensors<int64>(TensorShape({4}), in GetNextTestCases()
283 CreateTensors<int64>(TensorShape({4}), in GetNextTestCases()
287 CreateTensors<int64>(TensorShape({4}), in GetNextTestCases()
419 CreateTensors<int64>(TensorShape({4}), in IteratorSaveAndRestoreTestCases()
424 CreateTensors<int64>(TensorShape({4}), in IteratorSaveAndRestoreTestCases()
453 CreateTensors<int64>(TensorShape({4}), in IteratorSaveAndRestoreTestCases()
[all …]
Dparallel_map_dataset_op_test.cc274 CreateTensors<int64>(TensorShape{}, {{0}, {6}, {12}, {18}}), in GetNextTestCases()
278 CreateTensors<int64>(TensorShape{}, {{0}, {6}, {12}, {18}}), in GetNextTestCases()
282 CreateTensors<int64>(TensorShape{}, {{0}, {12}, {24}, {36}}), in GetNextTestCases()
286 CreateTensors<int64>(TensorShape{}, {{0}, {6}, {12}, {18}}), in GetNextTestCases()
290 CreateTensors<int64>(TensorShape{}, {{0}, {12}, {24}, {36}}), in GetNextTestCases()
295 CreateTensors<int64>(TensorShape{}, {{0}, {12}, {24}, {36}}), in GetNextTestCases()
375 CreateTensors<int64>(TensorShape{}, {{0}, {6}, {12}, {18}}), in IteratorSaveAndRestoreTestCases()
380 CreateTensors<int64>(TensorShape{}, {{0}, {6}, {12}, {18}}), in IteratorSaveAndRestoreTestCases()
385 CreateTensors<int64>(TensorShape{}, {{0}, {12}, {24}, {36}}), in IteratorSaveAndRestoreTestCases()
390 CreateTensors<int64>(TensorShape{}, {{0}, {6}, {12}, {18}}), in IteratorSaveAndRestoreTestCases()
[all …]
Dmap_dataset_op_test.cc97 CreateTensors<int64>(TensorShape({}), {{0}, {12}, {24}, {36}})}, in GetNextTestCases()
100 CreateTensors<int64>(TensorShape({2}), {{20, 14}, {8, 2}})}, in GetNextTestCases()
103 CreateTensors<int64>(TensorShape({}), {{0}, {12}, {24}, {36}})}}; in GetNextTestCases()
169 CreateTensors<int64>(TensorShape({}), {{0}, {12}, {24}, {36}})}, in IteratorSaveAndRestoreTestCases()
173 CreateTensors<int64>(TensorShape({2}), {{20, 14}, {8, 2}})}, in IteratorSaveAndRestoreTestCases()
177 CreateTensors<int64>(TensorShape({}), {{0}, {12}, {24}, {36}})}}; in IteratorSaveAndRestoreTestCases()
Dtake_dataset_op_test.cc63 CreateTensors<int64>(TensorShape({}), {{0}, {1}, {2}, {3}})}, in GetNextTestCases()
66 CreateTensors<int64>( in GetNextTestCases()
71 CreateTensors<int64>( in GetNextTestCases()
174 CreateTensors<int64>(TensorShape({}), {{0}, {1}, {2}, {3}})}, in ITERATOR_PREFIX_TEST_P()
178 CreateTensors<int64>( in ITERATOR_PREFIX_TEST_P()
184 CreateTensors<int64>( in ITERATOR_PREFIX_TEST_P()
Dconcatenate_dataset_op_test.cc28 /*components=*/CreateTensors<int64>(TensorShape{2, 2}, in SameShapeConcatenateDatasetParams()
32 /*components=*/CreateTensors<int64>(TensorShape{2, 2}, in SameShapeConcatenateDatasetParams()
66 /*components=*/CreateTensors<int64>(TensorShape{2, 2}, {{1, 2, 3, 4}}), in DifferentDtypeConcatenateDatasetParams()
70 CreateTensors<double>(TensorShape{2, 2}, {{1.0, 2.0, 3.0, 4.0}}), in DifferentDtypeConcatenateDatasetParams()
84 CreateTensors<int64>(TensorShape({2}), {{1, 2}, in GetNextTestCases()
205 CreateTensors<int64>(TensorShape({2}), {{1, 2}, in IteratorSaveAndRestoreTestCases()
Dbatch_dataset_op_test.cc122 CreateTensors<int64>(TensorShape({4}), in GetNextTestCases()
126 CreateTensors<int64>(TensorShape({4}), in GetNextTestCases()
136 CreateTensors<int64>(TensorShape({3}), in GetNextTestCases()
142 CreateTensors<int64>(TensorShape({10}), in GetNextTestCases()
250 CreateTensors<int64>(TensorShape({4}), in IteratorSaveAndRestoreTestCases()
255 CreateTensors<int64>(TensorShape({4}), in IteratorSaveAndRestoreTestCases()
Dreduce_dataset_op_test.cc100 /*initial_state=*/CreateTensors<int64>(TensorShape({}), {{1}}), in ReduceDatasetParams1()
121 /*initial_state=*/CreateTensors<int64>(TensorShape({}), {{1}, {1}}), in ReduceDatasetParams2()
139 /*initial_state=*/CreateTensors<int64>(TensorShape({}), {{1}, {3}}), in ReduceDatasetParams3()
156 CreateTensors<int64>(TensorShape({}), {{46}})}, in GetNextTestCases()
159 CreateTensors<int64>(TensorShape({}), in GetNextTestCases()
164 CreateTensors<int64>(TensorShape{}, {{1}, {3}})}}; in GetNextTestCases()
Dtext_line_dataset_op_test.cc145 CreateTensors<tstring>(TensorShape({}), {{"hello world"}, in GetNextTestCases()
151 CreateTensors<tstring>(TensorShape({}), {{"hello world"}, in GetNextTestCases()
157 CreateTensors<tstring>(TensorShape({}), {{"hello world"}, in GetNextTestCases()
223 CreateTensors<tstring>(TensorShape({}), {{"hello world"}, in IteratorSaveAndRestoreTestCases()
230 CreateTensors<tstring>(TensorShape({}), {{"hello world"}, in IteratorSaveAndRestoreTestCases()
237 CreateTensors<tstring>(TensorShape({}), {{"hello world"}, in IteratorSaveAndRestoreTestCases()
Dfinalize_dataset_op_test.cc219 CreateTensors<int64>(TensorShape({}), {{0}, {3}, {6}, {9}})}, in GetNextTestCases()
222 CreateTensors<int64>(TensorShape({}), {{0}, {3}, {6}, {9}})}, in GetNextTestCases()
225 CreateTensors<int64>(TensorShape({}), {{0}, {3}, {6}, {9}})}, in GetNextTestCases()
228 CreateTensors<int64>(TensorShape({}), {{0}, {3}, {6}, {9}})}, in GetNextTestCases()
231 CreateTensors<int64>(TensorShape({}), {{0}, {3}, {6}, {9}})}, in GetNextTestCases()
234 CreateTensors<int64>(TensorShape({}), {{0}, {3}, {6}, {9}})}}; in GetNextTestCases()
Dfixed_length_record_dataset_op_test.cc178 CreateTensors<tstring>(TensorShape({}), in GetNextTestCases()
181 CreateTensors<tstring>(TensorShape({}), in GetNextTestCases()
184 CreateTensors<tstring>(TensorShape({}), in GetNextTestCases()
252 CreateTensors<tstring>(TensorShape({}), in IteratorSaveAndRestoreTestCases()
256 CreateTensors<tstring>(TensorShape({}), in IteratorSaveAndRestoreTestCases()
260 CreateTensors<tstring>(TensorShape({}), in IteratorSaveAndRestoreTestCases()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddirected_interleave_dataset_op_test.cc301 /*expected_outputs=*/{CreateTensors<int64>( in GetNextTestCases()
304 /*expected_outputs=*/{CreateTensors<int64>( in GetNextTestCases()
307 /*expected_outputs=*/{CreateTensors<int64>( in GetNextTestCases()
310 /*expected_outputs=*/{CreateTensors<int64>(TensorShape({}), in GetNextTestCases()
313 /*expected_outputs=*/{CreateTensors<int64>(TensorShape({}), in GetNextTestCases()
316 /*expected_outputs=*/{CreateTensors<int64>(TensorShape({}), {})}}, in GetNextTestCases()
318 /*expected_outputs=*/{CreateTensors<int64>( in GetNextTestCases()
321 /*expected_outputs=*/{CreateTensors<int64>( in GetNextTestCases()
385 CreateTensors<int64>(TensorShape{}, {{0}, {10}, {1}, {11}, {2}, {12}}), in IteratorSaveAndRestoreTestCases()
390 CreateTensors<int64>(TensorShape{}, {{0}, {10}, {1}, {11}, {12}}), in IteratorSaveAndRestoreTestCases()
[all …]
Dparallel_interleave_dataset_op_test.cc376 CreateTensors<int64>(TensorShape{1}, in GetNextTestCases()
381 CreateTensors<int64>(TensorShape{1}, in GetNextTestCases()
386 CreateTensors<int64>(TensorShape{1}, in GetNextTestCases()
391 CreateTensors<int64>(TensorShape{1}, in GetNextTestCases()
396 CreateTensors<tstring>( in GetNextTestCases()
402 CreateTensors<tstring>(TensorShape{1}, {}), in GetNextTestCases()
481 CreateTensors<int64>(TensorShape{1}, in IteratorSaveAndRestoreTestCases()
487 CreateTensors<int64>(TensorShape{1}, in IteratorSaveAndRestoreTestCases()
493 CreateTensors<int64>(TensorShape{1}, in IteratorSaveAndRestoreTestCases()
499 CreateTensors<int64>(TensorShape{1}, in IteratorSaveAndRestoreTestCases()
[all …]
Dunique_dataset_op_test.cc153 CreateTensors<int64>(TensorShape({1}), in GetNextTestCases()
157 CreateTensors<int64>(TensorShape({1}), in GetNextTestCases()
161 CreateTensors<int64>(TensorShape({1}), {{1}})}, in GetNextTestCases()
164 CreateTensors<int64>(TensorShape({1}), {})}, in GetNextTestCases()
167 CreateTensors<tstring>(TensorShape({1}), {{"one"}, in GetNextTestCases()
252 CreateTensors<int64>(TensorShape({1}), in IteratorSaveAndRestoreTestCases()
257 CreateTensors<int64>(TensorShape({1}), in IteratorSaveAndRestoreTestCases()
Dmap_and_batch_dataset_op_test.cc247 CreateTensors<int64>(TensorShape({2}), {{0, 4}, {8, 12}})}, in GetNextTestCases()
250 CreateTensors<int64>(TensorShape({2}), {{0, 4}, {8, 12}})}, in GetNextTestCases()
258 CreateTensors<int64>(TensorShape({2}), {{0, 4}, {8, 12}})}, in GetNextTestCases()
261 CreateTensors<int64>(TensorShape({2}), {{0, 8}, {16, 24}})}, in GetNextTestCases()
379 CreateTensors<int64>(TensorShape({2}), {{0, 4}, {8, 12}})}, in IteratorSaveAndRestoreTestCases()
383 CreateTensors<int64>(TensorShape({2}), {{0, 4}, {8, 12}})}, in IteratorSaveAndRestoreTestCases()
393 CreateTensors<int64>(TensorShape({2}), {{0, 4}, {8, 12}})}, in IteratorSaveAndRestoreTestCases()
397 CreateTensors<int64>(TensorShape({2}), {{0, 8}, {16, 24}})}, in IteratorSaveAndRestoreTestCases()
/external/tensorflow/tensorflow/core/data/
Dsplit_utils_test.cc61 CheckOutput(&split_provider, CreateTensors<int64>(TensorShape({}), {}))); in TEST()
67 CreateTensors<int64>(TensorShape({}), {{0}}))); in TEST()
73 &split_provider, CreateTensors<int64>(TensorShape({}), {{0}, {1}, {2}}))); in TEST()
79 CreateTensors<int64>(TensorShape({}), {{0}, {1}, {2}, {3}}); in TEST()
99 CreateTensors<int64>(TensorShape({}), {{0}, {2}}))); in TEST()
106 CreateTensors<int64>(TensorShape({}), {{1}, {3}}))); in TEST()
113 CreateTensors<int64>(TensorShape({}), {{1}, {4}}))); in TEST()
120 CheckOutput(&split_provider, CreateTensors<int64>(TensorShape({}), {}))); in TEST()
126 CreateTensors<int64>(TensorShape({}), {{1}, {4}}); in TEST()
Dcompression_utils_test.cc40 CreateTensors<int64>(TensorShape{1}, {{1}}), // int64 in TestCases()
41 CreateTensors<int64>(TensorShape{1}, {{1}, {2}}), // multiple int64 in TestCases()
42 CreateTensors<tstring>(TensorShape{1}, {{"a"}, {"b"}}), // tstring in TestCases()

12