Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Doptions_dataset_op_test.cc119 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_INT64})); in TEST_F()
Dmap_dataset_op_test.cc148 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_INT64})); in DATASET_CARDINALITY_TEST_P()
Dtake_dataset_op_test.cc132 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_INT64})); in DATASET_CARDINALITY_TEST_P()
Dbatch_dataset_op_test.cc211 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_INT64})); in DATASET_CARDINALITY_TEST_P()
Dskip_dataset_op_test.cc178 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_INT64})); in DATASET_CARDINALITY_TEST_P()
Dflat_map_dataset_op_test.cc185 TF_ASSERT_OK(CheckIteratorOutputDtypes(dataset_params.output_dtypes())); in TEST_F()
Dtext_line_dataset_op_test.cc201 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_STRING})); in TEST_F()
Dprefetch_dataset_op_test.cc277 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_INT64})); in DATASET_CARDINALITY_TEST_P()
Dfilter_dataset_op_test.cc226 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_INT64})); in DATASET_CARDINALITY_TEST_P()
Dtf_record_dataset_op_test.cc229 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_STRING})); in TEST_F()
Dshard_dataset_op_test.cc274 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_INT64})); in DATASET_CARDINALITY_TEST_P()
Dfixed_length_record_dataset_op_test.cc227 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_STRING})); in TEST_F()
Dparallel_batch_dataset_op_test.cc373 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_INT64})); in DATASET_CARDINALITY_TEST_P()
Dtensor_dataset_op_test.cc189 TF_ASSERT_OK(CheckIteratorOutputDtypes(dataset_params.output_dtypes())); in TEST_F()
Dparallel_map_dataset_op_test.cc351 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_INT64})); in DATASET_CARDINALITY_TEST_P()
Dcache_dataset_ops_test.cc258 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_INT64})); in DATASET_CARDINALITY_TEST_P()
Dshuffle_dataset_op_test.cc440 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_INT64})); in DATASET_CARDINALITY_TEST_P()
Dwindow_dataset_op_test.cc389 TF_ASSERT_OK(CheckIteratorOutputDtypes(dataset_params.output_dtypes())); in DATASET_CARDINALITY_TEST_P()
Dparallel_interleave_dataset_op_test.cc665 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_INT64})); in DATASET_CARDINALITY_TEST_P()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dassert_next_dataset_op_test.cc178 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_INT64})); in TEST_F()
Ddirected_interleave_dataset_op_test.cc362 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_INT64})); in TEST_F()
Dmap_and_batch_dataset_op_test.cc338 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_INT64})); in DATASET_CARDINALITY_TEST_P()
Dparallel_interleave_dataset_op_test.cc457 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_INT64})); in DATASET_CARDINALITY_TEST_P()
/external/tensorflow/tensorflow/core/data/
Ddataset_test_base.h659 Status CheckIteratorOutputDtypes(
Ddataset_test_base.cc708 Status DatasetOpsTestBase::CheckIteratorOutputDtypes( in CheckIteratorOutputDtypes() function in tensorflow::data::DatasetOpsTestBase