15#ifndef O2_GPUTPCNNCLUSTERIZER_H
16#define O2_GPUTPCNNCLUSTERIZER_H
float * mOutputDataReg2_32
OrtDataType::Float16_t * mInputData_16
int32_t mNnClusterizerSizeInputTime
uint32_t mNnClusterizerRowTimeSize
int32_t mNnClusterizerSizeInputRow
int32_t mNnClusterizerSizeInputPad
void * setIOPointers(void *)
uint32_t mNnClusterizerFullPadSize
int32_t mNnClusterizerBatchedMode
int32_t mNnClusterizerTotalClusters
int32_t mNnClusterizerElementSize
int32_t mNnClusterizerModelReg1NumOutputNodes
int32_t mNnClusterizerVerbosity
OrtDataType::Float16_t * mOutputDataReg2_16
float * mModelProbabilities_32
int32_t mNnClusterizerUseCfRegression
int32_t mNnClusterizerModelClassNumOutputNodes
int32_t mNnClusterizerBoundaryFillValue
GPUTPCNNClusterizer()=default
int8_t mNnClusterizerAddIndexData
int8_t mNnSigmoidTrafoClassThreshold
void SetMaxData(const GPUTrackingInOutPointers &)
int32_t mNnInferenceInputDType
int32_t * mOutputDataClass
uint32_t mNnClusterizerRowTimeSizeFull
int32_t mNnInferenceOutputDType
float * mOutputDataReg1_32
uint32_t mNnClusterizerFullRowSize
void InitializeProcessor()
int32_t mNnClusterizerModelReg2NumOutputNodes
OrtDataType::Float16_t * mModelProbabilities_16
int32_t mNnClusterizerChargeArraySize
uint32_t mNnClusterizerFullTimeSize
int8_t mNnClusterizerSetDeconvolutionFlags
uint32_t mNnClusterizerPadTimeSize
OrtDataType::Float16_t * mOutputDataReg1_16
void RegisterMemoryAllocation()
int8_t mNnClusterizerUseClassification