/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
H A D | options_dataset_op_test.cc | 101 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
|
H A D | take_dataset_op_test.cc | 97 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
|
H A D | map_dataset_op_test.cc | 124 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
|
H A D | skip_dataset_op_test.cc | 151 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
|
H A D | flat_map_dataset_op_test.cc | 168 TF_ASSERT_OK(CheckDatasetOutputDtypes(dataset_params.output_dtypes())); in TEST_F()
|
H A D | prefetch_dataset_op_test.cc | 248 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
|
H A D | text_line_dataset_op_test.cc | 184 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_STRING})); in TEST_F()
|
H A D | tf_record_dataset_op_test.cc | 212 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_STRING})); in TEST_F()
|
H A D | shard_dataset_op_test.cc | 242 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
|
H A D | filter_dataset_op_test.cc | 201 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
|
H A D | fixed_length_record_dataset_op_test.cc | 210 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_STRING})); in TEST_F()
|
H A D | batch_dataset_op_test.cc | 175 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
|
H A D | parallel_batch_dataset_op_test.cc | 316 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
|
H A D | cache_dataset_ops_test.cc | 223 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
|
H A D | parallel_filter_dataset_op_test.cc | 334 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
|
H A D | tensor_dataset_op_test.cc | 172 TF_ASSERT_OK(CheckDatasetOutputDtypes(dataset_params.output_dtypes())); in TEST_F()
|
H A D | parallel_map_dataset_op_test.cc | 319 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
H A D | dataset_test_base.h | 656 Status CheckDatasetOutputDtypes(const DataTypeVector& expected_output_dtypes); 929 TF_ASSERT_OK(CheckDatasetOutputDtypes(test_case.expected_output_dtypes)); \ 988 TF_ASSERT_OK(CheckDatasetOutputDtypes(test_case.expected_output_dtypes)); \
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
H A D | assert_next_dataset_op_test.cc | 160 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
|
H A D | unique_dataset_op_test.cc | 198 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
|
H A D | assert_prev_dataset_op_test.cc | 215 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
|
H A D | io_ops_test.cc | 197 TF_ASSERT_OK(CheckDatasetOutputDtypes(dataset_params.output_dtypes())); in TEST_F()
|
H A D | map_and_batch_dataset_op_test.cc | 288 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
|
H A D | directed_interleave_dataset_op_test.cc | 355 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
|
H A D | parallel_interleave_dataset_op_test.cc | 427 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
|