ActsPlugins::EdgeClassificationBase Class Reference

#include <ActsPlugins/Gnn/Stages.hpp>

Inheritance diagram for ActsPlugins::EdgeClassificationBase:

Public Member Functions
virtual	~EdgeClassificationBase ()=default
virtual PipelineTensors	operator() (PipelineTensors tensors, const ExecutionContext &execContext={})=0
	Perform edge classification.

Constructor & Destructor Documentation

~EdgeClassificationBase()

virtual ActsPlugins::EdgeClassificationBase::~EdgeClassificationBase ( )

virtualdefault

Member Function Documentation

operator()()

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

pure virtual

Perform edge classification.

Parameters

tensors	Input pipeline tensors
execContext	Device & stream information

Returns

(node_features, edge_features, edge_index, edge_scores)

Implemented in ActsPlugins::OnnxEdgeClassifier, ActsPlugins::TensorRTEdgeClassifier, and ActsPlugins::TorchEdgeClassifier.