Skip to content

Commit ba59bea

Browse files
author
Changhwan Choi
committed
Resolved MegaLinter errors
1 parent ad57ddf commit ba59bea

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

‎Tools/ML/model.h‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424

2525
#include <onnxruntime_cxx_api.h>
2626

27+
#include <algorithm>
2728
#include <cstddef>
2829
#include <cstdint>
2930
#include <memory>
@@ -230,9 +231,9 @@ class OnnxModel
230231
return mSession;
231232
}
232233
int getNumInputNodes() const { return mInputShapes[0][1]; }
233-
std::vector<std::vector<int64_t>> getInputShapes() const { return mInputShapes; }
234+
const std::vector<std::vector<int64_t>>& getInputShapes() const { return mInputShapes; }
234235
int getNumOutputNodes() const { return mOutputShapes[0][1]; }
235-
std::vector<std::vector<int64_t>> getOutputShapes() const { return mOutputShapes; }
236+
const std::vector<std::vector<int64_t>>& getOutputShapes() const { return mOutputShapes; }
236237
uint64_t getValidityFrom() const { return validFrom; }
237238
uint64_t getValidityUntil() const { return validUntil; }
238239
void setActiveThreads(const int);

0 commit comments

Comments
 (0)