StageModelExecutor Class
Stage executor that runs a Model's inference pipeline as a runtime-graph stage. More...
Declaration
Included Headers
Base class
| class | StageExecutor |
|
Actor-style executor base class — implement to add a new runtime-graph stage. More... | |
Public Constructors Index
| StageModelExecutor (StageModelExecutorOptions opt) | |
|
Construct a StageModelExecutor from the given options. More... | |
Public Member Functions Index
| void | set_ports (const StagePorts &ports) override |
|
Bind the executor to its runtime ports. More... | |
| void | on_input (StageMsg &&msg, std::vector< StageOutMsg > &out) override |
|
Run the configured inference sub-stages on the incoming sample. More... | |
Private Member Attributes Index
| StageModelExecutorOptions | opt_ |
| PortId | out_port_ = kInvalidPort |
Description
Stage executor that runs a Model's inference pipeline as a runtime-graph stage.
Wraps the staged inference APIs (preprocess / MLA / box-decode) so a Model can be embedded directly in a runtime graph alongside other stages.
- See Also
- See Also
Definition at line 51 of file StageModelExecutor.h.
Public Constructors
StageModelExecutor()
| explicit |
Construct a StageModelExecutor from the given options.
Definition at line 54 of file StageModelExecutor.h.
Public Member Functions
on_input()
| virtual |
Run the configured inference sub-stages on the incoming sample.
Definition at line 59 of file StageModelExecutor.h.
set_ports()
| virtual |
Bind the executor to its runtime ports.
Definition at line 57 of file StageModelExecutor.h.
Private Member Attributes
opt_
|
Definition at line 62 of file StageModelExecutor.h.
out_port_
|
Definition at line 63 of file StageModelExecutor.h.
The documentation for this class was generated from the following file:
Generated via doxygen2docusaurus 2.0.0 by Doxygen 1.9.8.