SiSSKJr SSKrSSKJr Sr"SS\5r"SS\5r"S S \5r "S S \5r g) ) annotationsN)OpRunc4[U[R5(a%URS:wa[ SU<S35eUSn[ U5n[R "UR5UnU(aU*n[R"X@4US9n[R"XUS9nU(a([R"XRS9n[R"XbS9n[R"U5n[R"XWUS9n[R"XgUS9n X4$)zSee function `_kneighbors_reduce_func `_. zk must be an integer not .raxis) isinstancenpndarraysize RuntimeErrorintindicesshapelexsorttake_along_axisfliparangetake) Xkr largestind_axissorted_indices sorted_valuesarktopk_sorted_indicestopk_sorted_valuess Y/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/reference/ops/op_topk.pytopk_sorted_implementationr! s !RZZ  66Q;!:1%qAB B aD AAzz!''"4(H9ZZ D9N&&qtDM; 9 ))A,C''.DA$?  22c\rSrSrSSjrSrg) _CommonTopK%cUSnUS:aUOU[UR5-n[XX45upgXgR[R 54$)Runtime for operator *TopK*. The implementation is not the most efficient as it sorts everything then extracts the top *k* values. .. warning:: ONNX specifications may be imprecise in case of negative value for axis. The implementation follows what `onnxruntime` does in `top_k.cc `_. r)lenrr!astyper int64)selfdatainkr rrsortsortis r _common_run_CommonTopK._common_run&sK Fqyttc$**o'=0$H ll288,--r"N)r)__name__ __module__ __qualname____firstlineno__r0__static_attributes__r2r"r r$r$%s.r"r$c\rSrSrSSjrSrg)TopK_18Nc.[RXU/US9$r'rr$r0)r+r,rr s r _run TopK_1._run9s&&tA3T&BBr"r2)NNr3r4r5r6r>r7r2r"r r9r98 Cr"r9c\rSrSrSSjrSrg)TopK_10HNc*[RXX#S9$r<r=)r+r,r-r s r r> TopK_10._runIs&&t3&BBr"r2)Nr@r2r"r rCrCHrAr"rCc$\rSrSrSSjrSrg)TopK_11XNcNUS;a [S5e[RXX#US9$)aRuntime for operator *TopK*. The implementation is not the most efficient as it sorts everything then extracts the top *k* values. .. warning:: ONNX specifications may be imprecise in case of negative value for axis. The implementation follows what `onnxruntime` does in `top_k.cc `_. )Trz.TopK does not implement anything for sorted=0.)r r)rr$r0)r+r,r-r rsorteds r r> TopK_11._runYs0(  "OP P&&t37&SSr"r2)NNNr@r2r"r rHrHXs  Tr"rH) __future__rnumpyr onnx.reference.op_runrr!r$r9rCrHr2r"r rPsK#'34.%.& C[ C Ck C TkTr"