Searched refs:PrefetchAutotuner (Results 1 – 4 of 4) sorted by relevance
24 TEST(PrefetchAutotuner, Disabled) { in TEST() argument25 PrefetchAutotuner t(2); in TEST()34 TEST(PrefetchAutotuner, Enabled) { in TEST() argument35 PrefetchAutotuner t(PrefetchAutotuner::kAutoTune); in TEST()59 TEST(PrefetchAutotuner, EnabledSteady) { in TEST() argument60 PrefetchAutotuner t(PrefetchAutotuner::kAutoTune); in TEST()
21 PrefetchAutotuner::PrefetchAutotuner(int64 initial_buffer_size) in PrefetchAutotuner() function in tensorflow::data::PrefetchAutotuner36 void PrefetchAutotuner::RecordConsumption(size_t current_buffer_size) { in RecordConsumption()
40 class PrefetchAutotuner {44 explicit PrefetchAutotuner(int64 initial_buffer_size);
372 PrefetchAutotuner auto_tuner_ GUARDED_BY(mu_);388 buffer_size >= 0 || buffer_size == PrefetchAutotuner::kAutoTune, in MakeDataset()391 if (buffer_size == PrefetchAutotuner::kAutoTune) { in MakeDataset()