ReUseX  0.0.5
3D Point Cloud Processing for Building Reuse
Loading...
Searching...
No Matches
reusex::vision::tensor_rt::TensorRT::Engine Class Referenceabstract

#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: