Searched refs:would_exceed_memory_limit (Results 1 – 1 of 1) sorted by relevance
136 bool would_exceed_memory_limit(std::size_t bytes) const in would_exceed_memory_limit() function in tensorflow::__anon07039e130111::StagingMap262 while (would_exceed_memory_limit(tuple_bytes)) { in put_incomplete()361 while (would_exceed_memory_limit(tuple_bytes) || is_capacity_full()) { in put()