Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Dprefetch_autotuner_test.cc26 EXPECT_EQ(2, t.buffer_limit()); in TEST()
31 EXPECT_EQ(2, t.buffer_limit()); in TEST()
36 EXPECT_EQ(1, t.buffer_limit()); in TEST()
38 EXPECT_EQ(1, t.buffer_limit()); in TEST()
40 EXPECT_EQ(1, t.buffer_limit()); in TEST()
42 EXPECT_EQ(2, t.buffer_limit()); in TEST()
44 EXPECT_EQ(2, t.buffer_limit()); in TEST()
46 EXPECT_EQ(2, t.buffer_limit()); in TEST()
48 EXPECT_EQ(4, t.buffer_limit()); in TEST()
50 EXPECT_EQ(4, t.buffer_limit()); in TEST()
[all …]
Dprefetch_dataset_op.cc115 auto_tuner_.buffer_limit() != 0) { in GetNextInternal()
136 DCHECK_EQ(auto_tuner_.buffer_limit(), 0); in GetNextInternal()
147 static_cast<float>(auto_tuner_.buffer_limit())); in GetNextInternal()
238 static_cast<float>(auto_tuner_.buffer_limit())}); in Consume()
244 static_cast<float>(auto_tuner_.buffer_limit())); in Consume()
288 while (!cancelled_ && buffer_.size() >= auto_tuner_.buffer_limit()) { in PrefetchThread()
Dprefetch_autotuner.h46 int64 buffer_limit() const { return buffer_limit_; } in buffer_limit() function