/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
H A D | options_dataset_op_test.cc | 107 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
|
H A D | map_dataset_op_test.cc | 130 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
|
H A D | range_dataset_op_test.cc | 86 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in DATASET_OUTPUT_DTYPES_TEST_P()
|
H A D | skip_dataset_op_test.cc | 157 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
|
H A D | flat_map_dataset_op_test.cc | 174 TF_ASSERT_OK(CheckDatasetOutputShapes(dataset_params.output_shapes())); in TEST_F()
|
H A D | prefetch_dataset_op_test.cc | 254 TF_ASSERT_OK(CheckDatasetOutputShapes(dataset_params.output_shapes())); in TEST_F()
|
H A D | text_line_dataset_op_test.cc | 190 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
|
H A D | tf_record_dataset_op_test.cc | 218 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
|
H A D | shard_dataset_op_test.cc | 248 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
|
H A D | fixed_length_record_dataset_op_test.cc | 216 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
|
H A D | tensor_dataset_op_test.cc | 178 TF_ASSERT_OK(CheckDatasetOutputShapes(dataset_params.output_shapes())); in TEST_F()
|
H A D | parallel_map_dataset_op_test.cc | 325 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
|
H A D | shuffle_dataset_op_test.cc | 412 TF_ASSERT_OK(CheckDatasetOutputShapes(dataset_params.output_shapes())); in TEST_F()
|
H A D | window_dataset_op_test.cc | 357 TF_ASSERT_OK(CheckDatasetOutputShapes(dataset_params.output_shapes())); in TEST_F()
|
H A D | parallel_interleave_dataset_op_test.cc | 747 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({1})})); in TEST_F()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
H A D | assert_next_dataset_op_test.cc | 166 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
|
H A D | unique_dataset_op_test.cc | 204 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({1})})); in TEST_F()
|
H A D | assert_prev_dataset_op_test.cc | 221 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
|
H A D | directed_interleave_dataset_op_test.cc | 361 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
|
H A D | parallel_interleave_dataset_op_test.cc | 433 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({1})})); in TEST_F()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
H A D | dataset_test_base.h | 659 Status CheckDatasetOutputShapes( 949 TF_ASSERT_OK(CheckDatasetOutputShapes(test_case.expected_output_shapes)); \
|
H A D | dataset_test_base.cc | 699 Status DatasetOpsTestBase::CheckDatasetOutputShapes( in CheckDatasetOutputShapes() function in tensorflow::data::DatasetOpsTestBase
|