15#ifndef O2_GPUTPCNNCLUSTERIZER_H
16#define O2_GPUTPCNNCLUSTERIZER_H
float * mOutputDataReg2_32
OrtDataType::Float16_t * mInputData_16
int mNnClusterizerSizeInputPad
int mNnClusterizerBatchedMode
int mNnClusterizerModelReg1NumOutputNodes
int mNnInferenceOutputDType
int mNnInferenceInputDType
int mNnClusterizerElementSize
void * setIOPointers(void *)
int mNnClusterizerModelClassNumOutputNodes
OrtDataType::Float16_t * mOutputDataReg2_16
float * mModelProbabilities_32
bool mNnClusterizerAddIndexData
int mNnClusterizerModelReg2NumOutputNodes
int mNnClusterizerTotalClusters
int mNnClusterizerSizeInputRow
GPUTPCNNClusterizer()=default
int mNnClusterizerUseCfRegression
int mNnClusterizerVerbosity
void SetMaxData(const GPUTrackingInOutPointers &)
bool mNnSigmoidTrafoClassThreshold
float * mOutputDataReg1_32
void InitializeProcessor()
int mNnClusterizerSizeInputTime
OrtDataType::Float16_t * mModelProbabilities_16
int mNnClusterizerBoundaryFillValue
OrtDataType::Float16_t * mOutputDataReg1_16
void RegisterMemoryAllocation()