Skip to content

Commit

Permalink
[NoACR][MachineLearning.Inference] Fix bug using wrong handle
Browse files Browse the repository at this point in the history
This patch fixes the bug which use wrong handle.
GetTensorsCount should use TensorsInfo handle instead of TensorsData handle.

Signed-off-by: Yelin Jeong <[email protected]>
  • Loading branch information
niley7464 authored and again4you committed Dec 5, 2023
1 parent 9f11206 commit 7facd0f
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@ private TensorsData(IntPtr handle, TensorsInfo info, bool isFetch = false, bool

/* Set count */
int count = 0;
ret = Interop.Util.GetTensorsCount(_handle, out count);
ret = Interop.Util.GetTensorsCount(_tensorsInfo.GetTensorsInfoHandle(), out count);
NNStreamer.CheckException(ret, "unable to get the count of TensorsData");

_dataList = new ArrayList(count);
Expand Down

0 comments on commit 7facd0f

Please sign in to comment.