Searched refs:SmallBatchToSpaceLayerDataset (Results 1 – 3 of 3) sorted by relevance
110 class SmallBatchToSpaceLayerDataset final : public BatchToSpaceLayerDataset113 SmallBatchToSpaceLayerDataset() in SmallBatchToSpaceLayerDataset() function
108 … framework::DatasetMode::PRECOMMIT, combine(combine(datasets::SmallBatchToSpaceLayerDataset(), fra… in TEST_SUITE()125 … framework::DatasetMode::PRECOMMIT, combine(combine(datasets::SmallBatchToSpaceLayerDataset(), fra… in TEST_SUITE()
110 … framework::DatasetMode::PRECOMMIT, combine(combine(datasets::SmallBatchToSpaceLayerDataset(), fra… in TEST_SUITE()127 … framework::DatasetMode::PRECOMMIT, combine(combine(datasets::SmallBatchToSpaceLayerDataset(), fra… in TEST_SUITE()