Searched refs:TotalMaximumBufferedBytes (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | model_test.cc | 64 EXPECT_EQ(async_interleave_many->TotalMaximumBufferedBytes(), 0); in TEST_P()
67 EXPECT_EQ(async_interleave_many->TotalMaximumBufferedBytes(), in TEST_P()
134 EXPECT_EQ(async_known_many->TotalMaximumBufferedBytes(), 0); in TEST_P()
137 EXPECT_EQ(async_known_many->TotalMaximumBufferedBytes(), in TEST_P()
408 EXPECT_EQ(node->TotalMaximumBufferedBytes(), 0); in TEST()
414 EXPECT_EQ(node->TotalMaximumBufferedBytes(), 60); in TEST()
420 EXPECT_EQ(node->TotalMaximumBufferedBytes(), 45); in TEST()
428 EXPECT_EQ(node->TotalMaximumBufferedBytes(), 48); in TEST()
438 EXPECT_EQ(node->TotalMaximumBufferedBytes(), 51); in TEST()
448 EXPECT_EQ(node->TotalMaximumBufferedBytes(), 49.5); in TEST()
[all …]
|
| D | model.h | 370 double TotalMaximumBufferedBytes() const TF_LOCKS_EXCLUDED(mu_);
770 double TotalMaximumBufferedBytes(std::shared_ptr<Node> node);
|
| D | model.cc | 1209 double Node::TotalMaximumBufferedBytes() const { in TotalMaximumBufferedBytes() function in tensorflow::data::model::Node
1723 return all_max || TotalMaximumBufferedBytes(snapshot) > ram_budget; in ShouldStop()
1870 TotalMaximumBufferedBytes(snapshot) > in OptimizeHillClimb()
1924 double Model::TotalMaximumBufferedBytes(std::shared_ptr<Node> node) { in TotalMaximumBufferedBytes() function in tensorflow::data::model::Model
1925 return node->TotalMaximumBufferedBytes(); in TotalMaximumBufferedBytes()
|