ActsPlugins::OnnxEdgeClassifier Class Reference

ONNX-based edge classifier for GNN pipeline. More...

#include <ActsPlugins/Gnn/OnnxEdgeClassifier.hpp>

Inheritance diagram for ActsPlugins::OnnxEdgeClassifier:

Collaboration diagram for ActsPlugins::OnnxEdgeClassifier:

Classes
struct	Config
	Configuration for OnnxEdgeClassifier. More...

Public Member Functions
	OnnxEdgeClassifier (const Config &cfg, std::unique_ptr< const Acts::Logger > logger)
	Constructor.
Config	config () const
	Get configuration.
PipelineTensors	operator() (PipelineTensors tensors, const ExecutionContext &execContext={}) override
	Perform edge classification.

Detailed Description

ONNX-based edge classifier for GNN pipeline.

Constructor & Destructor Documentation

OnnxEdgeClassifier()

ActsPlugins::OnnxEdgeClassifier::OnnxEdgeClassifier	(	const Config &	cfg,
		std::unique_ptr< const Acts::Logger >	logger )

Constructor.

Parameters

cfg	Configuration parameters
logger	Logger instance

Member Function Documentation

config()

Config ActsPlugins::OnnxEdgeClassifier::config

(

)

const

Get configuration.

Returns

Configuration parameters

operator()()

PipelineTensors ActsPlugins::OnnxEdgeClassifier::operator() ( PipelineTensors tensors, const ExecutionContext & execContext = {} )

overridevirtual

Perform edge classification.

Parameters

tensors	Input pipeline tensors
execContext	Device & stream information

Returns

(node_features, edge_features, edge_index, edge_scores)

Implements ActsPlugins::EdgeClassificationBase.