Home
last modified time | relevance | path

Searched refs:CheckDatasetOutputDtypes (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dmap_dataset_op_test.cc124 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
Dtake_dataset_op_test.cc97 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
Dskip_dataset_op_test.cc150 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
Dbatch_dataset_op_test.cc170 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
Dflat_map_dataset_op_test.cc149 TF_ASSERT_OK(CheckDatasetOutputDtypes(dataset_params.output_dtypes())); in TEST_F()
Dprefetch_dataset_op_test.cc249 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
Dtext_line_dataset_op_test.cc183 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_STRING})); in TEST_F()
Dfilter_dataset_op_test.cc199 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
Dtf_record_dataset_op_test.cc211 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_STRING})); in TEST_F()
Dshard_dataset_op_test.cc242 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
Dfixed_length_record_dataset_op_test.cc209 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_STRING})); in TEST_F()
Dparallel_batch_dataset_op_test.cc259 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
Ddataset_test_base.h610 Status CheckDatasetOutputDtypes(const DataTypeVector& expected_output_dtypes);
876 TF_ASSERT_OK(CheckDatasetOutputDtypes(test_case.expected_output_dtypes)); \
935 TF_ASSERT_OK(CheckDatasetOutputDtypes(test_case.expected_output_dtypes)); \
Dtensor_dataset_op_test.cc166 TF_ASSERT_OK(CheckDatasetOutputDtypes(dataset_params.output_dtypes())); in TEST_F()
Dcache_dataset_ops_test.cc218 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
Dparallel_map_dataset_op_test.cc320 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
Dshuffle_dataset_op_test.cc401 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in DATASET_TYPE_STRING_TEST_P()
Dwindow_dataset_op_test.cc346 TF_ASSERT_OK(CheckDatasetOutputDtypes(dataset_params.output_dtypes())); in TEST_F()
Dparallel_interleave_dataset_op_test.cc625 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
Ddataset_test_base.cc684 Status DatasetOpsTestBase::CheckDatasetOutputDtypes( in CheckDatasetOutputDtypes() function in tensorflow::data::DatasetOpsTestBase
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dassert_next_dataset_op_test.cc160 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
Dunique_dataset_op_test.cc197 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
Ddirected_interleave_dataset_op_test.cc273 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
Dmap_and_batch_dataset_op_test.cc288 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()
Dparallel_interleave_dataset_op_test.cc427 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F()