Skip to main content

StageModelExecutor Class

Stage executor that runs a Model's inference pipeline as a runtime-graph stage. More...

Declaration​

class simaai::neat::graph::nodes::StageModelExecutor { ... }

Included Headers​

Base class​

classStageExecutor

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​

voidset_ports (const StagePorts &ports) override

Bind the executor to its runtime ports. More...

voidon_input (StageMsg &&msg, std::vector< StageOutMsg > &out) override

Run the configured inference sub-stages on the incoming sample. More...

Private Member Attributes Index​

StageModelExecutorOptionsopt_
PortIdout_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

StageModelExecutorNode

See Also

StageExecutor

Definition at line 51 of file StageModelExecutor.h.

Public Constructors​

StageModelExecutor()​

simaai::neat::graph::nodes::StageModelExecutor::StageModelExecutor (StageModelExecutorOptions opt)
explicit

Construct a StageModelExecutor from the given options.

Definition at line 54 of file StageModelExecutor.h.

Public Member Functions​

on_input()​

void simaai::neat::graph::nodes::StageModelExecutor::on_input (StageMsg && msg, std::vector< StageOutMsg > & out)
virtual

Run the configured inference sub-stages on the incoming sample.

Definition at line 59 of file StageModelExecutor.h.

set_ports()​

void simaai::neat::graph::nodes::StageModelExecutor::set_ports (const StagePorts & ports)
virtual

Bind the executor to its runtime ports.

Definition at line 57 of file StageModelExecutor.h.

Private Member Attributes​

opt_​

StageModelExecutorOptions simaai::neat::graph::nodes::StageModelExecutor::opt_

Definition at line 62 of file StageModelExecutor.h.

out_port_​

PortId simaai::neat::graph::nodes::StageModelExecutor::out_port_ = kInvalidPort

Definition at line 63 of file StageModelExecutor.h.

63 PortId out_port_ = kInvalidPort;

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


Generated via doxygen2docusaurus 2.0.0 by Doxygen 1.9.1.