Searched refs:expected_output_shapes (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib.cc | 302 std::vector<PartialTensorShape> expected_output_shapes(expected_max_outputs); in Execute() local
305 auto result = Join(expected_output_shapes, status); in Execute()
341 const std::vector<PartialTensorShape>& expected_output_shapes, in Join() argument
392 if (expected_output_shapes[i].IsFullyDefined()) { in Join()
395 absl::Span<const int64>(expected_output_shapes[i].dim_sizes()), in Join()
|
| D | parallel_device_lib.h | 130 const std::vector<PartialTensorShape>& expected_output_shapes,
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | tflite_convert_test.py | 58 expected_output_shapes=None): argument
73 if expected_output_shapes:
75 self.assertEqual(output_shapes, expected_output_shapes)
506 expected_output_shapes=[[1, 1], [1, 2]])
516 expected_output_shapes=[[1, 1], [1, 2]])
|
| /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/kernels/data/ |
| D | dataset_test_base.h | 425 std::vector<PartialTensorShape> expected_output_shapes; member
448 std::vector<PartialTensorShape> expected_output_shapes; member
614 const std::vector<PartialTensorShape>& expected_output_shapes);
625 const std::vector<PartialTensorShape>& expected_output_shapes);
896 TF_ASSERT_OK(CheckDatasetOutputShapes(test_case.expected_output_shapes)); \
954 TF_ASSERT_OK(CheckIteratorOutputShapes(test_case.expected_output_shapes)); \
|
| D | dataset_test_base.cc | 692 const std::vector<PartialTensorShape>& expected_output_shapes) { in CheckDatasetOutputShapes() argument
694 expected_output_shapes)); in CheckDatasetOutputShapes()
711 const std::vector<PartialTensorShape>& expected_output_shapes) { in CheckIteratorOutputShapes() argument
713 expected_output_shapes)); in CheckIteratorOutputShapes()
|