We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 80984a7 commit 56a8c92Copy full SHA for 56a8c92
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