Searched refs:PlainTensorSliceDatasetParams (Results 1 – 1 of 1) sorted by relevance
28 TensorSliceDatasetParams PlainTensorSliceDatasetParams() { in PlainTensorSliceDatasetParams() function59 {/*dataset_params=*/PlainTensorSliceDatasetParams(), in GetNextTestCases()140 auto dataset_params = PlainTensorSliceDatasetParams(); in TEST_F()146 auto dataset_params = PlainTensorSliceDatasetParams(); in TEST_F()154 return {{PlainTensorSliceDatasetParams(), in DatasetOutputTypesTestCases()155 PlainTensorSliceDatasetParams().output_dtypes()}, in DatasetOutputTypesTestCases()165 return {{PlainTensorSliceDatasetParams(), in DATASET_OUTPUT_DTYPES_TEST_P()166 PlainTensorSliceDatasetParams().output_shapes()}, in DATASET_OUTPUT_DTYPES_TEST_P()176 return {{PlainTensorSliceDatasetParams(), /*expected_cardinality=*/2}, in DATASET_OUTPUT_SHAPES_TEST_P()185 return {{PlainTensorSliceDatasetParams(), in DATASET_CARDINALITY_TEST_P()[all …]