@@ -262,7 +262,7 @@ template std::vector<float> OrtModel::inference<OrtDataType::Float16_t, float>(s
262262template std::vector<OrtDataType::Float16_t> OrtModel::inference<OrtDataType::Float16_t, OrtDataType::Float16_t>(std::vector<OrtDataType::Float16_t>&);
263263
264264template <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
297297template <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
357357template <class I , class O >
358358std::vector<O> OrtModel::inference (std::vector<std::vector<I>>& inputs)
0 commit comments