ActsPlugins::GnnPipeline Class Reference

#include <ActsPlugins/Gnn/GnnPipeline.hpp>

Public Member Functions
	GnnPipeline (std::shared_ptr< GraphConstructionBase > graphConstructor, std::vector< std::shared_ptr< EdgeClassificationBase > > edgeClassifiers, std::shared_ptr< TrackBuildingBase > trackBuilder, std::unique_ptr< const Acts::Logger > logger)
std::vector< std::vector< int > >	run (std::vector< float > &features, const std::vector< std::uint64_t > &moduleIds, std::vector< int > &spacepointIDs, Device device, const GnnHook &hook={}, GnnTiming *timing=nullptr) const

Constructor & Destructor Documentation

GnnPipeline()

ActsPlugins::GnnPipeline::GnnPipeline	(	std::shared_ptr< GraphConstructionBase >	graphConstructor,
		std::vector< std::shared_ptr< EdgeClassificationBase > >	edgeClassifiers,
		std::shared_ptr< TrackBuildingBase >	trackBuilder,
		std::unique_ptr< const Acts::Logger >	logger )

Member Function Documentation

run()

std::vector< std::vector< int > > ActsPlugins::GnnPipeline::run	(	std::vector< float > &	features,
		const std::vector< std::uint64_t > &	moduleIds,
		std::vector< int > &	spacepointIDs,
		Device	device,
		const GnnHook &	hook = {},
		GnnTiming *	timing = nullptr ) const