From 32c3a5e1594871976077346a85462b5b22d53d2e Mon Sep 17 00:00:00 2001 From: dante <45801863+alexander-camuto@users.noreply.github.com> Date: Tue, 2 Apr 2024 21:37:20 +0100 Subject: [PATCH] fix: hold stacked outputs in a separate map --- examples/onnx/lstm_large/input.json | 13 ++++++++ examples/onnx/lstm_large/network.onnx | Bin 0 -> 1137918 bytes src/graph/model.rs | 44 ++++++++++++++++++++------ tests/integration_tests.rs | 5 +-- 4 files changed, 50 insertions(+), 12 deletions(-) create mode 100644 examples/onnx/lstm_large/input.json create mode 100644 examples/onnx/lstm_large/network.onnx diff --git a/examples/onnx/lstm_large/input.json b/examples/onnx/lstm_large/input.json new file mode 100644 index 000000000..2e54c296c --- /dev/null +++ b/examples/onnx/lstm_large/input.json @@ -0,0 +1,13 @@ +{ + "input_data": [ + [ + 0.8894134163856506, + 0.8894201517105103 + ] + ], + "output_data": [ + [ + 0.8436377 + ] + ] +} \ No newline at end of file diff --git a/examples/onnx/lstm_large/network.onnx b/examples/onnx/lstm_large/network.onnx new file mode 100644 index 0000000000000000000000000000000000000000..bff5a4fe38f510f2ac3fb73b0e4d2911522fc0aa GIT binary patch literal 1137918 zcmeFYc{JDG_dgm5Q4yLopj0wbhS%9D8c-ULL?xA>6p~7&%riyidCXKu=GWQgjLAHc zCe0;{O2d8inZEDuaPRtlziZv!x_?~O@;v8hKhN3E$JyuX@yX3Cz-4bPZ)0ukuyn4R zwA@B%xm5~E%JQ-&Ra?rp^EmR%I%REp!JcQHwb><;^OmNzW+o6-m*7S$O% z(`-EdlUaB;56YUDD9Q40>i)UP@o=jDxytj*;1N7wV}0J<#N6i0NfR4e`#*1>^k3c- z;MuLGufuvuK~?rIZMna+<(G2*lSY)|PfKDA53lJ7`%@Q9|GfQqZ7#i27H4fvoz&*g zH?_9>``=_XZiz)avrnF~GhmMD~D5GpV|DN*i zC@F9HJ4(t*e@97K`G13wvdX`q#A^QwC06@iC@HJ_pF!ylFaJ*{sjyb`FDR)f{1v6A ze?{pJB@fphS(1}g;NkisOmeb{JY0XINltd-{|G4Fzl(~T>?Rgcf9c5mwIjcD_P@yf zUj=I?&tF*hFVLFzFCmhXlV!m*=db<0;x*33^B-vc0hl}sxxWD>FZU0?;@1G3H z|Ji_l>HHgDtpEHQVDbw8hXDJ-&wm!LzkneBXBYkjFnOiF0#@?x0Q*DBGym^9`n)~s z-2Ahr6%_wDzM1vOzunmQ&l~^P^$ME=6#m0OYNGI;&fW9&cBf97oisUr=F|x@6L}U_ za{uFpL}%&!d75>8)UomWZ|3~ZoGbs2nfQarA5b^_@1FcC{14f9{@3&KtozIStV8;| znT(F9z0QR*GP-7_Ck;&PPMID*W470pwWNPlbAP8)KV^D;>&cV#EN9mlBt#o5_8Wn~k4M{lCq( z_O+RppYr0Rx!>^qq6I`PM_x_-nWx%iGg7%iG#_(0J*hkAtjha|H&Cf*MWtM$R%NTA zr&{+H2j)Jp_1NidUU}0?p)zpzH`eiGV9-x%wbLGlD~Eo{sy)ydrCIg5l{?dCS1ue_ zqh>~p)LdO8EBp3~tNB?;s`0L9f)>su_$;PcxlHag&g_b$MVrl`wZ>fSh|S*0hNL}} zTjO_v@rzk%PZ=94uS|=o)wG;b`^jNanc;Y-Qs}{=%J1GWXu@WvX5VP3cF4^dAT0p{ z>25Wnx8ArW!V60#y{za*AbdbUL4);!ZnvDqEqe@ID9;Pl$c z{i+=xWEcsYFP$pyYQ$o($WHp?>!HfNOK;(+uAQ31#=kA$e_x=qIqKB1iG(!wOnxr) z|6r?WFJ_CgZ=f$=}xW${!y8J;A@Ui#GQ^U*NwakX6y<{-*#*L_D6&5{hfAvxX%X ze>1$5H)*>)o~^z*2PPKogm;siVD>8rf5`SgYG@|Py8fo=YpWP5E|xJ$9Wv0QAqdhO zO3+c7pE|w@!jXAl^l}a0xv2%%%XC9_p4q6|=>fd+3-Fn-6dcXBC&Gsl=`{^|jD6l( zG5^OZC_Pw0j?Udg6xhLlJJK+Cs+xU0kp?1G9%Yu|#|aa0(|t)f#8Qg++L* zCkulnE8vA#BURDr!#@|X=+EU