Searched refs:expected_output_shapes (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | tensor_dataset_op_test.cc | 73 std::vector<PartialTensorShape> expected_output_shapes; member
147 test_case.expected_output_shapes, in TEST_P()
197 test_case.expected_output_shapes, in TEST_F()
223 test_case.expected_output_shapes, in TEST_F()
249 test_case.expected_output_shapes, in TEST_F()
275 test_case.expected_output_shapes, in TEST_F()
286 test_case.expected_output_shapes.size()); in TEST_F()
287 for (int i = 0; i < test_case.expected_output_shapes.size(); i++) { in TEST_F()
288 EXPECT_TRUE(test_case.expected_output_shapes[i].IsIdenticalTo( in TEST_F()
306 test_case.expected_output_shapes, in TEST_F()
[all …]
|
| D | take_dataset_op_test.cc | 66 std::vector<PartialTensorShape> expected_output_shapes; member
172 test_case.expected_output_shapes, in TEST_P()
223 test_case.expected_output_shapes, in TEST_F()
253 test_case.expected_output_shapes, in TEST_P()
284 test_case.expected_output_shapes, in TEST_P()
296 test_case.expected_output_shapes)); in TEST_P()
315 test_case.expected_output_shapes, in TEST_P()
345 test_case.expected_output_shapes, in TEST_F()
380 test_case.expected_output_shapes, in TEST_P()
418 test_case.expected_output_shapes, in TEST_P()
[all …]
|
| D | repeat_dataset_op_test.cc | 69 std::vector<PartialTensorShape> expected_output_shapes; member
146 test_case.expected_output_shapes, in TEST_P()
218 test_case.expected_output_shapes, in TEST_F()
248 test_case.expected_output_shapes, in TEST_P()
278 test_case.expected_output_shapes, in TEST_P()
289 test_case.expected_output_shapes)); in TEST_P()
308 test_case.expected_output_shapes, in TEST_P()
338 test_case.expected_output_shapes, in TEST_F()
373 test_case.expected_output_shapes, in TEST_P()
410 test_case.expected_output_shapes, in TEST_P()
[all …]
|
| D | zip_dataset_op_test.cc | 223 std::vector<PartialTensorShape> expected_output_shapes; in TEST_P() local
224 expected_output_shapes.reserve(num_tensors_per_slice); in TEST_P()
226 expected_output_shapes.emplace_back(test_case.expected_outputs[i].shape()); in TEST_P()
230 expected_output_shapes)); in TEST_P()
285 std::vector<PartialTensorShape> expected_output_shapes; in TEST_P() local
286 expected_output_shapes.reserve(num_tensors_per_slice); in TEST_P()
288 expected_output_shapes.emplace_back(test_case.expected_outputs[i].shape()); in TEST_P()
292 expected_output_shapes)); in TEST_P()
|
| D | concatenate_dataset_op_test.cc | 76 std::vector<PartialTensorShape> expected_output_shapes; member
150 test_case.expected_output_dtypes, test_case.expected_output_shapes, in CreateDatasetFromTestCase()
248 test_case.expected_output_shapes)); in TEST_P()
292 test_case.expected_output_shapes)); in TEST_P()
|
| D | sparse_tensor_slice_dataset_op_test.cc | 260 std::vector<PartialTensorShape> expected_output_shapes = { in TEST_F() local
264 expected_output_shapes)); in TEST_F()
397 std::vector<PartialTensorShape> expected_output_shapes = { in TEST_F() local
401 expected_output_shapes)); in TEST_F()
|
| D | tensor_slice_dataset_op_test.cc | 307 std::vector<PartialTensorShape> expected_output_shapes; in TEST_F() local
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | iterator_test.py | 867 expected_output_shapes): argument
881 self.assertEqual(expected_output_shapes,
|