File OnnxEdgeClassifier.hpp
-
namespace Acts
Note
This file is foreseen for the
Geometry
module to replaceExtent
-
class OnnxEdgeClassifier : public Acts::EdgeClassificationBase
- #include <Acts/Plugins/ExaTrkX/OnnxEdgeClassifier.hpp>
Public Functions
-
~OnnxEdgeClassifier()
-
inline Config config() const
-
virtual std::tuple<std::any, std::any, std::any> operator()(std::any nodes, std::any edges, int deviceHint = -1) override
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)
Private Functions
-
inline const auto &logger() const
Private Members
-
std::string m_inputNameEdges
-
std::string m_inputNameNodes
-
std::string m_outputNameScores
-
struct Config
- #include <Acts/Plugins/ExaTrkX/OnnxEdgeClassifier.hpp>
Public Members
-
float cut = 0.21
-
std::string modelPath
-
float cut = 0.21
-
~OnnxEdgeClassifier()
-
class OnnxEdgeClassifier : public Acts::EdgeClassificationBase
-
namespace Ort