We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 146a86e + b39b542 commit f92a6deCopy full SHA for f92a6de
1 file changed
src/backends/onnxruntime.c
@@ -571,6 +571,7 @@ int RAI_ModelRunORT(RAI_ModelRunCtx **mctxs, RAI_Error *error) {
571
ONNX_VALIDATE_STATUS(ort->GetDimensionsCount(info, &ndims))
572
int64_t dims[ndims];
573
ONNX_VALIDATE_STATUS(ort->GetDimensions(info, dims, ndims))
574
+ ort->ReleaseTensorTypeAndShapeInfo(info);
575
if (dims[0] != total_batch_size) {
576
RAI_SetError(error, RAI_EMODELRUN,
577
"ERR Model did not generate the expected batch size");
0 commit comments