Searched refs:elements_requested (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fifo_queue.cc | 130 tuple[0].dim_size(0) - attempt->elements_requested; in TryEnqueueMany()
138 --attempt->elements_requested; in TryEnqueueMany()
139 if (attempt->elements_requested == 0) { in TryEnqueueMany()
262 if (closed_ && queue_size < attempt->elements_requested) { in TryDequeueMany()
269 attempt->elements_requested - 1; in TryDequeueMany()
290 attempt->elements_requested = queue_size; in TryDequeueMany()
302 attempt->elements_requested, ", current size ", in TryDequeueMany()
318 ManyOutShape(i, attempt->elements_requested); in TryDequeueMany()
330 attempt->tuple[0].dim_size(0) - attempt->elements_requested; in TryDequeueMany()
337 --attempt->elements_requested; in TryDequeueMany()
[all …]
|
| D | priority_queue.cc | 159 tuple[0].dim_size(0) - attempt->elements_requested; in TryEnqueueMany()
181 --attempt->elements_requested; in TryEnqueueMany()
182 if (attempt->elements_requested == 0) { in TryEnqueueMany()
311 (!allow_small_batch && s < attempt->elements_requested)) { in TryDequeueMany()
315 attempt->elements_requested, ", current size ", s, ")")); in TryDequeueMany()
331 if (s < attempt->elements_requested) { in TryDequeueMany()
349 ManyOutShape(i, attempt->elements_requested); in TryDequeueMany()
361 attempt->tuple[0].dim_size(0) - attempt->elements_requested; in TryDequeueMany()
368 --attempt->elements_requested; in TryDequeueMany()
369 if (attempt->elements_requested == 0) { in TryDequeueMany()
|
| D | random_shuffle_queue_op.cc | 208 tuple[0].dim_size(0) - attempt->elements_requested; in TryEnqueueMany()
216 --attempt->elements_requested; in TryEnqueueMany()
217 if (attempt->elements_requested == 0) { in TryEnqueueMany()
341 if (closed_ && queue_size < attempt->elements_requested) { in TryDequeueMany()
347 attempt->elements_requested - 1; in TryDequeueMany()
368 attempt->elements_requested = queue_size; in TryDequeueMany()
380 attempt->elements_requested, ", current size ", in TryDequeueMany()
397 ManyOutShape(i, attempt->elements_requested); in TryDequeueMany()
409 attempt->tuple[0].dim_size(0) - attempt->elements_requested; in TryDequeueMany()
416 --attempt->elements_requested; in TryDequeueMany()
[all …]
|
| D | conditional_accumulator_base.h | 93 int elements_requested; member
101 Attempt(int elements_requested, DoneCallback done_callback, in Attempt()
104 : elements_requested(elements_requested), in Attempt()
|
| D | queue_base.h | 155 int32 elements_requested; member
166 Attempt(int32 elements_requested, DoneCallback done_callback, in Attempt()
169 : elements_requested(elements_requested), in Attempt()
|
| D | padding_fifo_queue.cc | 103 if (closed_ && queue_size < attempt->elements_requested) { in TryDequeueMany()
128 attempt->elements_requested = queue_size; in TryDequeueMany()
140 attempt->elements_requested, ", current size ", in TryDequeueMany()
154 --attempt->elements_requested; in TryDequeueMany()
156 if (attempt->elements_requested == 0) { in TryDequeueMany()
|
| D | conditional_accumulator_base.cc | 83 if (counter_ >= attempt->elements_requested) { in TryTakeGrad()
|