We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents dfaed79 + e74cedb commit b39b542Copy full SHA for b39b542
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