ReUseX::vision::tensor_rt::TensorRT::Engine Class Reference

#include <tensorrt.hpp>

Public Member Functions
virtual	~Engine ()=default
virtual bool	forward (const std::unordered_map< std::string, const void * > &bindings, void *stream=nullptr, void *input_consum_event=nullptr)=0
virtual int	index (const std::string &name)=0
virtual std::vector< int >	run_dims (const std::string &name)=0
virtual std::vector< int >	run_dims (int ibinding)=0
virtual std::vector< int >	static_dims (const std::string &name)=0
virtual std::vector< int >	static_dims (int ibinding)=0
virtual int	numel (const std::string &name)=0
virtual int	numel (int ibinding)=0
virtual int	num_bindings ()=0
virtual bool	is_input (int ibinding)=0
virtual bool	is_input (const std::string &name)=0
virtual bool	set_run_dims (const std::string &name, const std::vector< int > &dims)=0
virtual bool	set_run_dims (int ibinding, const std::vector< int > &dims)=0
virtual DType	dtype (const std::string &name)=0
virtual DType	dtype (int ibinding)=0
virtual bool	has_dynamic_dim ()=0
virtual void	print (const char *name="TensorRT-Engine")=0

Detailed Description

Definition at line 23 of file tensorrt.hpp.

Constructor & Destructor Documentation

~Engine()

virtual ReUseX::vision::tensor_rt::TensorRT::Engine::~Engine ( )

virtualdefault

Member Function Documentation

dtype() [1/2]

virtual DType ReUseX::vision::tensor_rt::TensorRT::Engine::dtype ( const std::string & name )

pure virtual

dtype() [2/2]

virtual DType ReUseX::vision::tensor_rt::TensorRT::Engine::dtype ( int ibinding )

pure virtual

forward()

virtual bool ReUseX::vision::tensor_rt::TensorRT::Engine::forward ( const std::unordered_map< std::string, const void * > & bindings, void * stream = nullptr, void * input_consum_event = nullptr )

pure virtual

has_dynamic_dim()

virtual bool ReUseX::vision::tensor_rt::TensorRT::Engine::has_dynamic_dim ( )

pure virtual

index()

virtual int ReUseX::vision::tensor_rt::TensorRT::Engine::index ( const std::string & name )

pure virtual

is_input() [1/2]

virtual bool ReUseX::vision::tensor_rt::TensorRT::Engine::is_input ( const std::string & name )

pure virtual

is_input() [2/2]

virtual bool ReUseX::vision::tensor_rt::TensorRT::Engine::is_input ( int ibinding )

pure virtual

num_bindings()

virtual int ReUseX::vision::tensor_rt::TensorRT::Engine::num_bindings ( )

pure virtual

numel() [1/2]

virtual int ReUseX::vision::tensor_rt::TensorRT::Engine::numel ( const std::string & name )

pure virtual

numel() [2/2]

virtual int ReUseX::vision::tensor_rt::TensorRT::Engine::numel ( int ibinding )

pure virtual

print()

virtual void ReUseX::vision::tensor_rt::TensorRT::Engine::print ( const char * name = "TensorRT-Engine" )

pure virtual

run_dims() [1/2]

virtual std::vector< int > ReUseX::vision::tensor_rt::TensorRT::Engine::run_dims ( const std::string & name )

pure virtual

run_dims() [2/2]

virtual std::vector< int > ReUseX::vision::tensor_rt::TensorRT::Engine::run_dims ( int ibinding )

pure virtual

set_run_dims() [1/2]

virtual bool ReUseX::vision::tensor_rt::TensorRT::Engine::set_run_dims ( const std::string & name, const std::vector< int > & dims )

pure virtual

set_run_dims() [2/2]

virtual bool ReUseX::vision::tensor_rt::TensorRT::Engine::set_run_dims ( int ibinding, const std::vector< int > & dims )

pure virtual

static_dims() [1/2]

virtual std::vector< int > ReUseX::vision::tensor_rt::TensorRT::Engine::static_dims ( const std::string & name )

pure virtual

static_dims() [2/2]

virtual std::vector< int > ReUseX::vision::tensor_rt::TensorRT::Engine::static_dims ( int ibinding )

pure virtual

---

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

• libs/reusex/include/ReUseX/vision/tensor_rt/common/tensorrt.hpp