Project
Loading...
Searching...
No Matches
Public Member Functions | Public Attributes | List of all members
o2::gpu::GPUTPCNNClusterizerHost Class Reference

#include <GPUTPCNNClusterizerHost.h>

Public Member Functions

 GPUTPCNNClusterizerHost ()=default
 
 GPUTPCNNClusterizerHost (const GPUSettingsProcessingNNclusterizer &, GPUTPCNNClusterizer &)
 
void networkInference (o2::ml::OrtModel model, GPUTPCNNClusterizer &clusterer, size_t size, float *output, int32_t dtype)
 

Public Attributes

std::unordered_map< std::string, std::string > OrtOptions
 
o2::ml::OrtModel model_class
 
o2::ml::OrtModel model_reg_1
 
o2::ml::OrtModel model_reg_2
 
std::vector< std::string > reg_model_paths
 

Detailed Description

Definition at line 36 of file GPUTPCNNClusterizerHost.h.

Constructor & Destructor Documentation

◆ GPUTPCNNClusterizerHost() [1/2]

o2::gpu::GPUTPCNNClusterizerHost::GPUTPCNNClusterizerHost ( )
default

◆ GPUTPCNNClusterizerHost() [2/2]

GPUTPCNNClusterizerHost::GPUTPCNNClusterizerHost ( const GPUSettingsProcessingNNclusterizer &  settings,
GPUTPCNNClusterizer &  clusterer 
)

Definition at line 22 of file GPUTPCNNClusterizerHost.cxx.

Member Function Documentation

◆ networkInference()

void GPUTPCNNClusterizerHost::networkInference ( o2::ml::OrtModel  model,
GPUTPCNNClusterizer &  clusterer,
size_t  size,
float *  output,
int32_t  dtype 
)

Definition at line 58 of file GPUTPCNNClusterizerHost.cxx.

Member Data Documentation

◆ model_class

o2::ml::OrtModel o2::gpu::GPUTPCNNClusterizerHost::model_class

Definition at line 45 of file GPUTPCNNClusterizerHost.h.

◆ model_reg_1

o2::ml::OrtModel o2::gpu::GPUTPCNNClusterizerHost::model_reg_1

Definition at line 45 of file GPUTPCNNClusterizerHost.h.

◆ model_reg_2

o2::ml::OrtModel o2::gpu::GPUTPCNNClusterizerHost::model_reg_2

Definition at line 45 of file GPUTPCNNClusterizerHost.h.

◆ OrtOptions

std::unordered_map<std::string, std::string> o2::gpu::GPUTPCNNClusterizerHost::OrtOptions

Definition at line 44 of file GPUTPCNNClusterizerHost.h.

◆ reg_model_paths

std::vector<std::string> o2::gpu::GPUTPCNNClusterizerHost::reg_model_paths

Definition at line 46 of file GPUTPCNNClusterizerHost.h.

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