Searched refs:TestDataset (Results 1 – 3 of 3) sorted by relevance
474 class TestDataset {478 TestDataset(std::unique_ptr<OpKernel> kernel_, in TestDataset() function538 std::unique_ptr<TestDataset>* dataset);543 const TestDataset& dataset,548 const TestDataset& dataset,
633 std::unique_ptr<TestDataset> dataset; in CheckSplitProviderFullIteration()648 std::unique_ptr<TestDataset> dataset; in CheckSplitProviderShardedIteration()793 std::unique_ptr<TestDataset>* dataset) { in MakeDataset()802 *dataset = std::make_unique<TestDataset>( in MakeDataset()858 const DatasetParams& dataset_params, const TestDataset& dataset, in MakeIterator()877 const DatasetParams& dataset_params, const TestDataset& dataset, in MakeIterator()
96 func TestDataset(t *testing.T) { func