Home
last modified time | relevance | path

Searched refs:batch_dataset_params (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dbatch_dataset_op_test.cc158 auto batch_dataset_params = BatchDatasetParams1(); in ITERATOR_GET_NEXT_TEST_P() local
159 TF_ASSERT_OK(Initialize(batch_dataset_params)); in ITERATOR_GET_NEXT_TEST_P()
160 TF_ASSERT_OK(CheckDatasetNodeName(batch_dataset_params.node_name())); in ITERATOR_GET_NEXT_TEST_P()
164 auto batch_dataset_params = BatchDatasetParams1(); in TEST_F() local
165 TF_ASSERT_OK(Initialize(batch_dataset_params)); in TEST_F()
167 params.op_version = batch_dataset_params.op_version(); in TEST_F()
173 auto batch_dataset_params = BatchDatasetParams1(); in TEST_F() local
174 TF_ASSERT_OK(Initialize(batch_dataset_params)); in TEST_F()
214 auto batch_dataset_params = BatchDatasetParams1(); in DATASET_CARDINALITY_TEST_P() local
215 TF_ASSERT_OK(Initialize(batch_dataset_params)); in DATASET_CARDINALITY_TEST_P()
[all …]
H A Dmap_dataset_op_test.cc55 auto batch_dataset_params = in MapDatasetParams2() local
64 std::move(batch_dataset_params), in MapDatasetParams2()