ActsPlugins::TensorRTEdgeClassifier Class Reference

#include <ActsPlugins/Gnn/TensorRTEdgeClassifier.hpp>

Inheritance diagram for ActsPlugins::TensorRTEdgeClassifier:

Collaboration diagram for ActsPlugins::TensorRTEdgeClassifier:

Classes
struct	Config

Public Member Functions
	TensorRTEdgeClassifier (const Config &cfg, std::unique_ptr< const Acts::Logger > logger)
	~TensorRTEdgeClassifier ()
Config	config () const
PipelineTensors	operator() (PipelineTensors tensors, const ExecutionContext &execContext={}) override
	Perform edge classification.
Public Member Functions inherited from ActsPlugins::EdgeClassificationBase
virtual	~EdgeClassificationBase ()=default

Constructor & Destructor Documentation

TensorRTEdgeClassifier()

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

~TensorRTEdgeClassifier()

ActsPlugins::TensorRTEdgeClassifier::~TensorRTEdgeClassifier

(

)

Member Function Documentation

config()

Config ActsPlugins::TensorRTEdgeClassifier::config

(

)

const

operator()()

PipelineTensors ActsPlugins::TensorRTEdgeClassifier::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.