ActsPlugins::GraphStoreHook Class Reference

Hook for storing graph data during GNN pipeline execution. More...

#include <ActsPlugins/Gnn/GraphStoreHook.hpp>

Inheritance diagram for ActsPlugins::GraphStoreHook:

Collaboration diagram for ActsPlugins::GraphStoreHook:

Public Types
using	Graph = std::pair<std::vector<std::int64_t>, std::vector<float>>
	Graph data structure (edges, edge features).

Public Member Functions
void	operator() (const PipelineTensors &tensors, const ExecutionContext &execCtx) const override
	Store graph data during pipeline execution.
const Graph &	storedGraph () const
	Get the stored graph data.

Detailed Description

Hook for storing graph data during GNN pipeline execution.

Member Function Documentation

operator()()

void ActsPlugins::GraphStoreHook::operator() ( const PipelineTensors & tensors, const ExecutionContext & execCtx ) const

overridevirtual

Store graph data during pipeline execution.

Parameters

tensors	Pipeline tensor data
execCtx	Execution context

Reimplemented from ActsPlugins::GnnHook.

storedGraph()

const Graph & ActsPlugins::GraphStoreHook::storedGraph

(

)

const

Get the stored graph data.

Returns

Reference to the stored graph