InDet::ScoredGraphHook Class Reference

#include <ActsGnnHookTool.h>

Inheritance diagram for InDet::ScoredGraphHook:

Collaboration diagram for InDet::ScoredGraphHook:

Public Member Functions
void	operator() (const ActsPlugins::PipelineTensors &tensors, const ActsPlugins::ExecutionContext &ctx) const override
const std::vector< float > &	getEdgeScores () const
const std::vector< int64_t > &	getEdgeIndex () const

Private Attributes
std::vector< float > m_edgeScores	ATLAS_THREAD_SAFE {}
std::vector< int64_t > m_edgeIndex	ATLAS_THREAD_SAFE {}

Detailed Description

Definition at line 16 of file ActsGnnHookTool.h.

Member Function Documentation

getEdgeIndex()

const std::vector< int64_t > & InDet::ScoredGraphHook::getEdgeIndex ( ) const

inline

Definition at line 41 of file ActsGnnHookTool.h.

                                                     {
          return m_edgeIndex;
      }

getEdgeScores()

const std::vector< float > & InDet::ScoredGraphHook::getEdgeScores ( ) const

inline

Definition at line 38 of file ActsGnnHookTool.h.

                                                    {
          return m_edgeScores;
      }

operator()()

void InDet::ScoredGraphHook::operator() ( const ActsPlugins::PipelineTensors & tensors, const ActsPlugins::ExecutionContext & ctx ) const

inlineoverride

Definition at line 18 of file ActsGnnHookTool.h.

                                                                             {
          
          ActsPlugins::ExecutionContext cpuCtx;
          cpuCtx.device = ActsPlugins::Device::Cpu();
          cpuCtx.stream = ctx.stream;
 
          if (tensors.edgeScores.has_value()) {
            const auto& scores = tensors.edgeScores.value();
            auto cpuScores = scores.clone(cpuCtx);
            m_edgeScores.assign(cpuScores.data(), cpuScores.data() + cpuScores.size());
          }
 
          if (tensors.edgeIndex.size() > 0) {
            const auto& edgeIndex = tensors.edgeIndex;
            auto cpuIndex = edgeIndex.clone(cpuCtx);
            m_edgeIndex.assign(cpuIndex.data(), cpuIndex.data() + cpuIndex.size());
          }
      }

Member Data Documentation

ATLAS_THREAD_SAFE [1/2]

std::vector<int64_t> m_edgeIndex InDet::ScoredGraphHook::ATLAS_THREAD_SAFE {}

mutableprivate

Definition at line 47 of file ActsGnnHookTool.h.

{};

ATLAS_THREAD_SAFE [2/2]

std::vector<float> m_edgeScores InDet::ScoredGraphHook::ATLAS_THREAD_SAFE {}

mutableprivate

Definition at line 46 of file ActsGnnHookTool.h.

{};

---

The documentation for this class was generated from the following file:

• ActsGnnHookTool.h