diff options
| author |  Eugene Brevdo <ebrevdo@gmail.com> | 2016-01-19 09:39:34 -0800 |
|---|---|---|
| committer |  Manjunath Kudlur <keveman@gmail.com> | 2016-01-20 07:47:54 -0800 |
| commit | e39629219e748b08177f2c457ba45d51f5370aae (patch) | |
| tree | b0861c366bbbdf8330bb1862aa5a32251c012e65 /tensorflow/core/kernels/queue_base.h | |
| parent | f592f23775e2a6ac75496829db5005d3bb70a3d2 (diff) | |
PaddingFIFOQueue is like FIFOQueue but allows dynamic shapes (using padding with DequeueMany)
Change: 112482056
Diffstat (limited to 'tensorflow/core/kernels/queue_base.h')
| -rw-r--r-- | tensorflow/core/kernels/queue_base.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/queue_base.h b/tensorflow/core/kernels/queue_base.h index 10a6d65417..1c0e377cda 100644 --- a/tensorflow/core/kernels/queue_base.h +++ b/tensorflow/core/kernels/queue_base.h @@ -146,6 +146,8 @@ class QueueBase : public QueueInterface { RunCallback run_callback; // must be run while holding mu_ bool is_cancelled; Tuple tuple; + // tuples is used by some implementations allowing dynamic shapes. + std::vector<Tuple> tuples; Attempt(int32 elements_requested, DoneCallback done_callback, OpKernelContext* context, CancellationToken cancellation_token, |