Searched refs:RecordConsumption (Results 1 – 4 of 4) sorted by relevance
27 t.RecordConsumption(0); in TEST()28 t.RecordConsumption(2); in TEST()29 t.RecordConsumption(0); in TEST()30 t.RecordConsumption(2); in TEST()37 t.RecordConsumption(0); // Expect buffer limit to stay the same. in TEST()39 t.RecordConsumption(1); in TEST()41 t.RecordConsumption(0); // Expect buffer limit to increase. in TEST()43 t.RecordConsumption(2); in TEST()45 t.RecordConsumption(1); in TEST()47 t.RecordConsumption(0); // Expect buffer limit to increase. in TEST()[all …]
48 void RecordConsumption(size_t current_buffer_size);49 void RecordEmpty() { RecordConsumption(0); } in RecordEmpty()
36 void PrefetchAutotuner::RecordConsumption(size_t current_buffer_size) { in RecordConsumption() function in tensorflow::data::PrefetchAutotuner
253 auto_tuner_.RecordConsumption(buffer_.size()); in Consume()