Struct caffe2_queue::RebatchingQueue

pub struct RebatchingQueue { /* private fields */ }

| TODO: This is a very naive implementation with | a single mutex. We can do the atomic index | + circular queue optimizations or pull | something more heavy-weight later

Implementations

impl RebatchingQueue

pub fn new(capacity: usize, num_blobs: usize) -> Self

pub fn can_read(&self) -> bool

pub fn dequeue( &mut self, context: &mut CPUContext, num_elements: usize, outputs: &Vec<*mut TensorCPU> ) -> bool

pub fn can_write(&self) -> bool

pub fn enqueue_one( &mut self, context: &mut CPUContext, inputs: &Vec<*const TensorCPU> ) -> bool

pub fn enqueue_many( &mut self, context: &mut CPUContext, inputs: &Vec<*const TensorCPU> ) -> bool

pub fn enqueue(&mut self, splitted_inputs: Vec<Vec<TensorCPU>>) -> bool

pub fn capacity(&self) -> usize

pub fn num_blobs(&self) -> usize

pub fn is_closed(&self) -> bool

pub fn close(&mut self)

Trait Implementations

impl Drop for RebatchingQueue

fn drop(&mut self)

Executes the destructor for this type.