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: