Home
last modified time | relevance | path

Searched refs:tuple_bytes (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dstage_op.cc44 std::size_t tuple_bytes = GetTupleBytes(*tuple); in Put() local
47 if (memory_limit_ > 0 && tuple_bytes > memory_limit_) { in Put()
51 tuple_bytes, in Put()
59 full_cond_var_.wait(lock, [tuple_bytes, this]() { in Put()
62 memory_limit_ > 0 ? !WouldExceedMemoryLimit(tuple_bytes) : true; in Put()
72 current_bytes_ += tuple_bytes; in Put()
Dmap_stage_op.cc258 std::size_t tuple_bytes = get_tuple_bytes(*tuple); in put_incomplete() local
259 TF_RETURN_IF_ERROR(check_memory_limit(tuple_bytes)); in put_incomplete()
262 while (would_exceed_memory_limit(tuple_bytes)) { in put_incomplete()
284 current_bytes_ += tuple_bytes; in put_incomplete()
304 current_bytes_ += tuple_bytes; in put_incomplete()
356 std::size_t tuple_bytes = get_tuple_bytes(*tuple); in put() local
358 TF_RETURN_IF_ERROR(check_memory_limit(tuple_bytes)); in put()
361 while (would_exceed_memory_limit(tuple_bytes) || is_capacity_full()) { in put()
369 current_bytes_ += tuple_bytes; in put()