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.8.