sample::Bindings Class Reference

Collaboration diagram for sample::Bindings:

Public Member Functions
void	addBinding (int b, const std::string &name, bool isInput, int volume, nvinfer1::DataType dataType, const std::string &fileName="")
void **	getDeviceBuffers ()
void	transferInputToDevice (TrtCudaStream &stream)
void	transferOutputToHost (TrtCudaStream &stream)
void	fill (int binding, const std::string &fileName)
void	fill (int binding)
void	dumpBindingDimensions (int binding, const nvinfer1::IExecutionContext &context, std::ostream &os) const
void	dumpBindingValues (int binding, std::ostream &os, const std::string &separator=" ") const
void	dumpInputs (const nvinfer1::IExecutionContext &context, std::ostream &os) const
void	dumpOutputs (const nvinfer1::IExecutionContext &context, std::ostream &os) const
void	dumpBindings (const nvinfer1::IExecutionContext &context, std::ostream &os) const
void	dumpBindings (const nvinfer1::IExecutionContext &context, bool(*predicate)(const Binding &b), std::ostream &os) const
std::unordered_map< std::string, int >	getInputBindings () const
std::unordered_map< std::string, int >	getOutputBindings () const
std::unordered_map< std::string, int >	getBindings () const
std::unordered_map< std::string, int >	getBindings (bool(*predicate)(const Binding &b)) const

Private Attributes
std::unordered_map< std::string, int >	mNames
std::vector< Binding >	mBindings
std::vector< void * >	mDevicePointers

Member Function Documentation

addBinding()

void sample::Bindings::addBinding ( int  b, const std::string &  name, bool  isInput, int  volume, nvinfer1::DataType  dataType, const std::string &  fileName = ""  )

inline

Here is the call graph for this function:

getDeviceBuffers()

void** sample::Bindings::getDeviceBuffers ( )

inline

transferInputToDevice()

void sample::Bindings::transferInputToDevice ( TrtCudaStream &  stream )

inline

transferOutputToHost()

void sample::Bindings::transferOutputToHost ( TrtCudaStream &  stream )

inline

fill() [1/2]

void sample::Bindings::fill ( int  binding, const std::string &  fileName  )

inline

Here is the caller graph for this function:

fill() [2/2]

void sample::Bindings::fill ( int  binding )

inline

dumpBindingDimensions()

void sample::Bindings::dumpBindingDimensions ( int  binding, const nvinfer1::IExecutionContext &  context, std::ostream &  os  ) const

inline

Here is the call graph for this function:

Here is the caller graph for this function:

dumpBindingValues()

void sample::Bindings::dumpBindingValues ( int  binding, std::ostream &  os, const std::string &  separator = " "  ) const

inline

Here is the caller graph for this function:

dumpInputs()

void sample::Bindings::dumpInputs ( const nvinfer1::IExecutionContext &  context, std::ostream &  os  ) const

inline

Here is the call graph for this function:

Here is the caller graph for this function:

dumpOutputs()

void sample::Bindings::dumpOutputs ( const nvinfer1::IExecutionContext &  context, std::ostream &  os  ) const

inline

Here is the call graph for this function:

Here is the caller graph for this function:

dumpBindings() [1/2]

void sample::Bindings::dumpBindings ( const nvinfer1::IExecutionContext &  context, std::ostream &  os  ) const

inline

Here is the caller graph for this function:

dumpBindings() [2/2]

void sample::Bindings::dumpBindings ( const nvinfer1::IExecutionContext &  context, bool(*)(const Binding &b)  predicate, std::ostream &  os  ) const

inline

Here is the call graph for this function:

getInputBindings()

std::unordered_map<std::string, int> sample::Bindings::getInputBindings ( ) const

inline

Here is the call graph for this function:

getOutputBindings()

std::unordered_map<std::string, int> sample::Bindings::getOutputBindings ( ) const

inline

Here is the call graph for this function:

Here is the caller graph for this function:

getBindings() [1/2]

std::unordered_map<std::string, int> sample::Bindings::getBindings ( ) const

inline

Here is the caller graph for this function:

getBindings() [2/2]

std::unordered_map<std::string, int> sample::Bindings::getBindings ( bool(*)(const Binding &b)  predicate ) const

inline

Member Data Documentation

mNames

std::unordered_map<std::string, int> sample::Bindings::mNames

private

mBindings

std::vector<Binding> sample::Bindings::mBindings

private

mDevicePointers

std::vector<void*> sample::Bindings::mDevicePointers

private

---

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

• sampleUtils.h