nndeploy C++ API  0.2.0
nndeploy C++ API
Public Member Functions | Public Attributes | List of all members
nndeploy::net::PipelineTensor Class Reference

#include <runtime.h>

Public Member Functions

 PipelineTensor ()
 
device::Tensorpop (Runtime *runtime)
 
void push (device::Tensor *tensor)
 
void setFinish ()
 
virtual ~PipelineTensor ()
 

Public Attributes

std::vector< Runtime * > consumers_
 
std::map< Runtime *, int > current_index_
 
std::condition_variable cv_
 
bool is_finish_ = false
 
std::mutex mutex_
 
std::vector< Runtime * > producers_
 
std::vector< device::Tensor * > tensors_
 

Detailed Description

Definition at line 21 of file runtime.h.

Constructor & Destructor Documentation

◆ PipelineTensor()

nndeploy::net::PipelineTensor::PipelineTensor ( )
inline

Definition at line 23 of file runtime.h.

◆ ~PipelineTensor()

virtual nndeploy::net::PipelineTensor::~PipelineTensor ( )
inlinevirtual

Definition at line 24 of file runtime.h.

Member Function Documentation

◆ pop()

device::Tensor* nndeploy::net::PipelineTensor::pop ( Runtime runtime)
inline

Definition at line 48 of file runtime.h.

◆ push()

void nndeploy::net::PipelineTensor::push ( device::Tensor tensor)
inline

Definition at line 35 of file runtime.h.

◆ setFinish()

void nndeploy::net::PipelineTensor::setFinish ( )
inline

Definition at line 42 of file runtime.h.

Member Data Documentation

◆ consumers_

std::vector<Runtime *> nndeploy::net::PipelineTensor::consumers_

Definition at line 27 of file runtime.h.

◆ current_index_

std::map<Runtime *, int> nndeploy::net::PipelineTensor::current_index_

Definition at line 32 of file runtime.h.

◆ cv_

std::condition_variable nndeploy::net::PipelineTensor::cv_

Definition at line 31 of file runtime.h.

◆ is_finish_

bool nndeploy::net::PipelineTensor::is_finish_ = false

Definition at line 33 of file runtime.h.

◆ mutex_

std::mutex nndeploy::net::PipelineTensor::mutex_

Definition at line 30 of file runtime.h.

◆ producers_

std::vector<Runtime *> nndeploy::net::PipelineTensor::producers_

Definition at line 26 of file runtime.h.

◆ tensors_

std::vector<device::Tensor *> nndeploy::net::PipelineTensor::tensors_

Definition at line 25 of file runtime.h.


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