Skip to content

Commit 70907aa

Browse files
committed
Fixing warnings (errors due to size_t)
1 parent 5f741fc commit 70907aa

File tree

2 files changed

+12
-12
lines changed

2 files changed

+12
-12
lines changed

Common/ML/include/ML/OrtInterface.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -98,10 +98,10 @@ class OrtModel
9898
std::vector<O> inference(std::vector<std::vector<I>>&);
9999

100100
template <class I, class O>
101-
void inference(I*, size_t, O*);
101+
void inference(I*, int64_t, O*);
102102

103103
template <class I, class O>
104-
void inference(I**, size_t, O*);
104+
void inference(I**, int64_t, O*);
105105

106106
void release();
107107

Common/ML/src/OrtInterface.cxx

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -262,7 +262,7 @@ template std::vector<float> OrtModel::inference<OrtDataType::Float16_t, float>(s
262262
template std::vector<OrtDataType::Float16_t> OrtModel::inference<OrtDataType::Float16_t, OrtDataType::Float16_t>(std::vector<OrtDataType::Float16_t>&);
263263

264264
template <class I, class O>
265-
void OrtModel::inference(I* input, size_t input_size, O* output)
265+
void OrtModel::inference(I* input, int64_t input_size, O* output)
266266
{
267267
// std::vector<std::string> providers = Ort::GetAvailableProviders();
268268
// for (const auto& provider : providers) {
@@ -289,13 +289,13 @@ void OrtModel::inference(I* input, size_t input_size, O* output)
289289
(pImplOrt->session)->Run(pImplOrt->runOptions, *pImplOrt->ioBinding);
290290
}
291291

292-
template void OrtModel::inference<OrtDataType::Float16_t, OrtDataType::Float16_t>(OrtDataType::Float16_t*, size_t, OrtDataType::Float16_t*);
293-
template void OrtModel::inference<OrtDataType::Float16_t, float>(OrtDataType::Float16_t*, size_t, float*);
294-
template void OrtModel::inference<float, OrtDataType::Float16_t>(float*, size_t, OrtDataType::Float16_t*);
295-
template void OrtModel::inference<float, float>(float*, size_t, float*);
292+
template void OrtModel::inference<OrtDataType::Float16_t, OrtDataType::Float16_t>(OrtDataType::Float16_t*, int64_t, OrtDataType::Float16_t*);
293+
template void OrtModel::inference<OrtDataType::Float16_t, float>(OrtDataType::Float16_t*, int64_t, float*);
294+
template void OrtModel::inference<float, OrtDataType::Float16_t>(float*, int64_t, OrtDataType::Float16_t*);
295+
template void OrtModel::inference<float, float>(float*, int64_t, float*);
296296

297297
template <class I, class O>
298-
void OrtModel::inference(I** input, size_t input_size, O* output)
298+
void OrtModel::inference(I** input, int64_t input_size, O* output)
299299
{
300300
std::vector<Ort::Value> inputTensors(inputShapesCopy.size());
301301

@@ -349,10 +349,10 @@ void OrtModel::inference(I** input, size_t input_size, O* output)
349349
outputNamesChar.size());
350350
}
351351

352-
template void OrtModel::inference<OrtDataType::Float16_t, OrtDataType::Float16_t>(OrtDataType::Float16_t**, size_t, OrtDataType::Float16_t*);
353-
template void OrtModel::inference<OrtDataType::Float16_t, float>(OrtDataType::Float16_t**, size_t, float*);
354-
template void OrtModel::inference<float, OrtDataType::Float16_t>(float**, size_t, OrtDataType::Float16_t*);
355-
template void OrtModel::inference<float, float>(float**, size_t, float*);
352+
template void OrtModel::inference<OrtDataType::Float16_t, OrtDataType::Float16_t>(OrtDataType::Float16_t**, int64_t, OrtDataType::Float16_t*);
353+
template void OrtModel::inference<OrtDataType::Float16_t, float>(OrtDataType::Float16_t**, int64_t, float*);
354+
template void OrtModel::inference<float, OrtDataType::Float16_t>(float**, int64_t, OrtDataType::Float16_t*);
355+
template void OrtModel::inference<float, float>(float**, int64_t, float*);
356356

357357
template <class I, class O>
358358
std::vector<O> OrtModel::inference(std::vector<std::vector<I>>& inputs)

0 commit comments

Comments
 (0)