Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Dtensor_slice_dataset_op_test.cc28 TensorSliceDatasetParams PlainTensorSliceDatasetParams() { in PlainTensorSliceDatasetParams() function
59 {/*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 …]