HomeSort by relevance Sort by last modified time
    Searched refs:tuple_bytes (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
stage_op.cc 89 std::size_t tuple_bytes = GetTupleBytes(*tuple); local
92 if (memory_limit_ > 0 && tuple_bytes > memory_limit_) {
96 tuple_bytes,
104 full_cond_var_.wait(lock, [tuple_bytes, this]() {
107 memory_limit_ > 0 ? !WouldExceedMemoryLimit(tuple_bytes) : true;
117 current_bytes_ += tuple_bytes;
map_stage_op.cc 258 std::size_t tuple_bytes = get_tuple_bytes(*tuple); local
259 TF_RETURN_IF_ERROR(check_memory_limit(tuple_bytes));
262 while (would_exceed_memory_limit(tuple_bytes)) {
284 current_bytes_ += tuple_bytes;
304 current_bytes_ += tuple_bytes;
356 std::size_t tuple_bytes = get_tuple_bytes(*tuple); local
357 // Check that tuple_bytes fits within the memory limit
358 TF_RETURN_IF_ERROR(check_memory_limit(tuple_bytes));
361 while (would_exceed_memory_limit(tuple_bytes) || is_capacity_full()) {
369 current_bytes_ += tuple_bytes;
    [all...]

Completed in 473 milliseconds