diff --git a/Common/ML/include/ML/OrtInterface.h b/Common/ML/include/ML/OrtInterface.h index 892111ad9ffad..980d3023f825a 100644 --- a/Common/ML/include/ML/OrtInterface.h +++ b/Common/ML/include/ML/OrtInterface.h @@ -119,7 +119,7 @@ class OrtModel std::vector mInputNamesChar, mOutputNamesChar; std::vector mInputNames, mOutputNames; std::vector> mInputShapes, mOutputShapes, mInputShapesCopy, mOutputShapesCopy; // Input shapes - std::vector mInputSizePerNode, mOutputSizePerNode; // Output shapes + std::vector mInputSizePerNode, mOutputSizePerNode; // Output shapes int32_t mInputsTotal = 0, mOutputsTotal = 0; // Total number of inputs and outputs // Environment settings diff --git a/GPU/GPUTracking/TPCClusterFinder/GPUTPCNNClusterizerHost.h b/GPU/GPUTracking/TPCClusterFinder/GPUTPCNNClusterizerHost.h index cf5a877b09ef7..e659753f21d7d 100644 --- a/GPU/GPUTracking/TPCClusterFinder/GPUTPCNNClusterizerHost.h +++ b/GPU/GPUTracking/TPCClusterFinder/GPUTPCNNClusterizerHost.h @@ -58,8 +58,8 @@ class GPUTPCNNClusterizerHost const OrtMemoryInfo* getMockedMemoryInfo(); std::unordered_map mOrtOptions; - o2::ml::OrtModel mModelClass, mModelReg1, mModelReg2; // For splitting clusters - std::vector mModelsUsed = {false, false, false}; // 0: class, 1: reg_1, 2: reg_2 + o2::ml::OrtModel mModelClass, mModelReg1, mModelReg2; // For splitting clusters + std::vector mModelsUsed = {false, false, false}; // 0: class, 1: reg_1, 2: reg_2 int32_t mDeviceId = -1; std::shared_ptr mMockedAlloc = nullptr; }; // class GPUTPCNNClusterizerHost