Public Member Functions | |
RebatchingQueue (size_t capacity, size_t numBlobs) | |
bool | enqueueOne (CPUContext &context, const std::vector< const TensorCPU * > &inputs) |
bool | enqueueMany (CPUContext &context, const std::vector< const TensorCPU * > &inputs) |
bool | dequeue (CPUContext &context, size_t numElements, const std::vector< TensorCPU * > &outputs) |
size_t | capacity () const |
size_t | numBlobs () const |
bool | isClosed () const |
void | close () |
Definition at line 20 of file rebatching_queue.h.