Acts::OnnxEdgeClassifier Exception Reference

#include <acts/blob/sPHENIX/Plugins/ExaTrkX/include/Acts/Plugins/ExaTrkX/OnnxEdgeClassifier.hpp>

Inheritance diagram for Acts::OnnxEdgeClassifier:

Collaboration diagram for Acts::OnnxEdgeClassifier:

Classes
struct	Config

Public Member Functions
	OnnxEdgeClassifier (const Config &cfg, std::unique_ptr< const Logger > logger)
	~OnnxEdgeClassifier ()
std::tuple< std::any, std::any, std::any >	operator() (std::any nodes, std::any edges, int deviceHint=-1) override
Config	config () const
Public Member Functions inherited from Acts::EdgeClassificationBase
virtual	~EdgeClassificationBase ()=default

Private Member Functions
const auto &	logger () const

Private Attributes
std::unique_ptr< const Acts::Logger >	m_logger
Config	m_cfg
std::unique_ptr< Ort::Env >	m_env
std::unique_ptr< Ort::Session >	m_model
std::string	m_inputNameNodes
std::string	m_inputNameEdges
std::string	m_outputNameScores

Detailed Description

Definition at line 24 of file OnnxEdgeClassifier.hpp.

View newest version in sPHENIX GitHub at line 24 of file OnnxEdgeClassifier.hpp

Constructor & Destructor Documentation

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

Definition at line 20 of file OnnxEdgeClassifier.cpp.

View newest version in sPHENIX GitHub at line 20 of file OnnxEdgeClassifier.cpp

References m_cfg, m_env, m_inputNameEdges, m_inputNameNodes, m_model, m_outputNameScores, and Acts::OnnxEdgeClassifier::Config::modelPath.

Acts::OnnxEdgeClassifier::~OnnxEdgeClassifier

(

)

Definition at line 44 of file OnnxEdgeClassifier.cpp.

View newest version in sPHENIX GitHub at line 44 of file OnnxEdgeClassifier.cpp

Member Function Documentation

Config Acts::OnnxEdgeClassifier::config ( ) const

inline

Definition at line 37 of file OnnxEdgeClassifier.hpp.

View newest version in sPHENIX GitHub at line 37 of file OnnxEdgeClassifier.hpp

References m_cfg.

const auto& Acts::OnnxEdgeClassifier::logger ( ) const

inlineprivate

Definition at line 41 of file OnnxEdgeClassifier.hpp.

View newest version in sPHENIX GitHub at line 41 of file OnnxEdgeClassifier.hpp

References m_logger.

std::tuple< std::any, std::any, std::any > Acts::OnnxEdgeClassifier::operator() ( std::any  nodes, std::any  edges, int  deviceHint = -1  )

overridevirtual

Perform edge classification

Parameters

nodes	Node tensor with shape (n_nodes, n_node_features)
edges	Edge-index tensor with shape (2, n_edges)
deviceHint	Which GPU to pick. Not relevant for CPU-only builds

Returns

(node_tensor, edge_tensor, score_tensor)

Implements Acts::EdgeClassificationBase.

Definition at line 46 of file OnnxEdgeClassifier.cpp.

View newest version in sPHENIX GitHub at line 46 of file OnnxEdgeClassifier.cpp

References ACTS_DEBUG, Acts::OnnxEdgeClassifier::Config::cut, m_cfg, m_inputNameEdges, m_inputNameNodes, m_model, m_outputNameScores, testing::internal::move(), and runSessionWithIoBinding().

Here is the call graph for this function:

Member Data Documentation

Config Acts::OnnxEdgeClassifier::m_cfg

private

Definition at line 43 of file OnnxEdgeClassifier.hpp.

View newest version in sPHENIX GitHub at line 43 of file OnnxEdgeClassifier.hpp

Referenced by config(), OnnxEdgeClassifier(), and operator()().

std::unique_ptr<Ort::Env> Acts::OnnxEdgeClassifier::m_env

private

Definition at line 45 of file OnnxEdgeClassifier.hpp.

View newest version in sPHENIX GitHub at line 45 of file OnnxEdgeClassifier.hpp

Referenced by OnnxEdgeClassifier().

std::string Acts::OnnxEdgeClassifier::m_inputNameEdges

private

Definition at line 49 of file OnnxEdgeClassifier.hpp.

View newest version in sPHENIX GitHub at line 49 of file OnnxEdgeClassifier.hpp

Referenced by OnnxEdgeClassifier(), and operator()().

std::string Acts::OnnxEdgeClassifier::m_inputNameNodes

private

Definition at line 48 of file OnnxEdgeClassifier.hpp.

View newest version in sPHENIX GitHub at line 48 of file OnnxEdgeClassifier.hpp

Referenced by OnnxEdgeClassifier(), and operator()().

std::unique_ptr<const Acts::Logger> Acts::OnnxEdgeClassifier::m_logger

private

Definition at line 40 of file OnnxEdgeClassifier.hpp.

View newest version in sPHENIX GitHub at line 40 of file OnnxEdgeClassifier.hpp

Referenced by logger().

std::unique_ptr<Ort::Session> Acts::OnnxEdgeClassifier::m_model

private

Definition at line 46 of file OnnxEdgeClassifier.hpp.

View newest version in sPHENIX GitHub at line 46 of file OnnxEdgeClassifier.hpp

Referenced by OnnxEdgeClassifier(), and operator()().

std::string Acts::OnnxEdgeClassifier::m_outputNameScores

private

Definition at line 50 of file OnnxEdgeClassifier.hpp.

View newest version in sPHENIX GitHub at line 50 of file OnnxEdgeClassifier.hpp

Referenced by OnnxEdgeClassifier(), and operator()().

---

The documentation for this exception was generated from the following files:

• acts/blob/sPHENIX/Plugins/ExaTrkX/include/Acts/Plugins/ExaTrkX/OnnxEdgeClassifier.hpp
• acts/blob/sPHENIX/Plugins/ExaTrkX/src/OnnxEdgeClassifier.cpp