Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
caffe2::RebatchingQueue Class Reference

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 ()

Detailed Description

Definition at line 36 of file rebatching_queue.h.

The documentation for this class was generated from the following files: