Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 2 additions & 7 deletions Common/ML/include/ML/OrtInterface.h
Original file line number Diff line number Diff line change
Expand Up @@ -34,10 +34,7 @@ struct MemoryInfo;
struct Env;
} // namespace Ort

namespace o2
{

namespace ml
namespace o2::ml
{

class OrtModel
Expand Down Expand Up @@ -127,8 +124,6 @@ class OrtModel
std::string printShape(const std::vector<std::vector<int64_t>>&, std::vector<std::string>&);
};

} // namespace ml

} // namespace o2
} // namespace o2::ml

#endif // O2_ML_ORTINTERFACE_H
6 changes: 3 additions & 3 deletions Common/ML/src/OrtInterface.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -287,9 +287,9 @@ std::vector<O> OrtModel::inference(std::vector<I>& input)
return outputValuesVec;
}

template std::vector<float> OrtModel::inference<float, float>(std::vector<float>&);
template std::vector<float> OrtModel::inference<OrtDataType::Float16_t, float>(std::vector<OrtDataType::Float16_t>&);
template std::vector<OrtDataType::Float16_t> OrtModel::inference<OrtDataType::Float16_t, OrtDataType::Float16_t>(std::vector<OrtDataType::Float16_t>&);
template std::vector<float> o2::ml::OrtModel::inference<float, float>(std::vector<float>&);
template std::vector<float> o2::ml::OrtModel::inference<OrtDataType::Float16_t, float>(std::vector<OrtDataType::Float16_t>&);
template std::vector<OrtDataType::Float16_t> o2::ml::OrtModel::inference<OrtDataType::Float16_t, OrtDataType::Float16_t>(std::vector<OrtDataType::Float16_t>&);

template <class I, class O>
void OrtModel::inference(I* input, int64_t input_size, O* output)
Expand Down
2 changes: 0 additions & 2 deletions GPU/GPUTracking/TPCClusterFinder/GPUTPCNNClusterizerHost.h
Original file line number Diff line number Diff line change
Expand Up @@ -20,8 +20,6 @@
#include <vector>
#include "ML/OrtInterface.h"

using namespace o2::ml;

class OrtMemoryInfo;
class OrtAllocator;
struct MockedOrtAllocator;
Expand Down