Searched refs:expected_output_shapes (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/python/ |
| D | tflite_convert_test.py | 59 expected_output_shapes=None): argument
74 if expected_output_shapes:
76 self.assertEqual(output_shapes, expected_output_shapes)
499 expected_output_shapes=[[1, 1], [1, 2]])
509 expected_output_shapes=[[1, 1], [1, 2]])
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib.cc | 316 std::vector<PartialTensorShape> expected_output_shapes(expected_max_outputs); in Execute() local
319 auto result = Join(expected_output_shapes, status); in Execute()
377 const std::vector<PartialTensorShape>& expected_output_shapes, in Join() argument
428 if (expected_output_shapes[i].IsFullyDefined()) { in Join()
431 absl::Span<const int64>(expected_output_shapes[i].dim_sizes()), in Join()
|
| D | parallel_device_lib.h | 135 const std::vector<PartialTensorShape>& expected_output_shapes,
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | iterator_test.py | 788 expected_output_shapes=[[]])))
792 expected_output_shapes): argument
805 self.assertEqual(expected_output_shapes,
816 expected_output_shapes=[[1]])))
820 expected_output_shapes): argument
839 self.assertEqual(expected_output_shapes,
860 expected_output_shapes={
867 expected_output_shapes): argument
886 self.assertEqual(expected_output_shapes,
|
| /external/tensorflow/tensorflow/core/data/ |
| D | dataset_test_base.h | 461 std::vector<PartialTensorShape> expected_output_shapes; member
484 std::vector<PartialTensorShape> expected_output_shapes; member
650 const std::vector<PartialTensorShape>& expected_output_shapes);
664 const std::vector<PartialTensorShape>& expected_output_shapes);
939 TF_ASSERT_OK(CheckDatasetOutputShapes(test_case.expected_output_shapes)); \
997 TF_ASSERT_OK(CheckIteratorOutputShapes(test_case.expected_output_shapes)); \
|
| D | dataset_test_base.cc | 690 const std::vector<PartialTensorShape>& expected_output_shapes) { in CheckDatasetOutputShapes() argument
692 expected_output_shapes)); in CheckDatasetOutputShapes()
716 const std::vector<PartialTensorShape>& expected_output_shapes) { in CheckIteratorOutputShapes() argument
718 expected_output_shapes)); in CheckIteratorOutputShapes()
|