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 | 158 tuple[0].dim_size(0) - attempt->elements_requested; in TryEnqueueMany()
180 --attempt->elements_requested; in TryEnqueueMany()
181 if (attempt->elements_requested == 0) { in TryEnqueueMany()
310 (!allow_small_batch && s < attempt->elements_requested)) { in TryDequeueMany()
314 attempt->elements_requested, ", current size ", s, ")")); in TryDequeueMany()
330 if (s < attempt->elements_requested) { in TryDequeueMany()
348 ManyOutShape(i, attempt->elements_requested); in TryDequeueMany()
360 attempt->tuple[0].dim_size(0) - attempt->elements_requested; in TryDequeueMany()
367 --attempt->elements_requested; in TryDequeueMany()
368 if (attempt->elements_requested == 0) { in TryDequeueMany()
|
| D | random_shuffle_queue_op.cc | 207 tuple[0].dim_size(0) - attempt->elements_requested; in TryEnqueueMany()
215 --attempt->elements_requested; in TryEnqueueMany()
216 if (attempt->elements_requested == 0) { in TryEnqueueMany()
340 if (closed_ && queue_size < attempt->elements_requested) { in TryDequeueMany()
346 attempt->elements_requested - 1; in TryDequeueMany()
367 attempt->elements_requested = queue_size; in TryDequeueMany()
379 attempt->elements_requested, ", current size ", in TryDequeueMany()
396 ManyOutShape(i, attempt->elements_requested); in TryDequeueMany()
408 attempt->tuple[0].dim_size(0) - attempt->elements_requested; in TryDequeueMany()
415 --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 | 159 int32 elements_requested; member
170 Attempt(int32 elements_requested, DoneCallback done_callback, in Attempt()
173 : 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()
|