Searched refs:TwoDimsSparseTensorSliceDatasetParams (Results 1 – 1 of 1) sorted by relevance
73 SparseTensorSliceDatasetParams TwoDimsSparseTensorSliceDatasetParams() { in TwoDimsSparseTensorSliceDatasetParams() function117 return {{/*dataset_params=*/TwoDimsSparseTensorSliceDatasetParams(), in GetNextTestCases()193 auto dataset_params = TwoDimsSparseTensorSliceDatasetParams(); in TEST_F()199 auto dataset_params = TwoDimsSparseTensorSliceDatasetParams(); in TEST_F()206 return {{/*dataset_params=*/TwoDimsSparseTensorSliceDatasetParams(), in DatasetOutputDtypesTestCases()222 return {{/*dataset_params=*/TwoDimsSparseTensorSliceDatasetParams(), in DATASET_OUTPUT_DTYPES_TEST_P()246 return {{/*dataset_params=*/TwoDimsSparseTensorSliceDatasetParams(), in DATASET_OUTPUT_SHAPES_TEST_P()262 return {{/*dataset_params=*/TwoDimsSparseTensorSliceDatasetParams(), in DATASET_CARDINALITY_TEST_P()278 return {{/*dataset_params=*/TwoDimsSparseTensorSliceDatasetParams(), in ITERATOR_OUTPUT_DTYPES_TEST_P()301 auto dataset_params = TwoDimsSparseTensorSliceDatasetParams(); in ITERATOR_OUTPUT_SHAPES_TEST_P()[all …]