hvSSKrSSKJr Sr"SS\5r"SS\5r"SS \5r"S S \5rg) N)OpRunc[U[R5(a%URS:wa[ SU<S35eUSn[ U5n[ UR5S:XaUS:Xa[R"URS5SS2S4nUS:XaC[R"XUS- S9nUSS2SU24nUU[R"XU454nOD[R"U*X!S- S9nUSS2SU24nUU[R"XU4*54nXU4nXe4$[R"XS9n[R"XS9nU(a([R"XRS9n[R"XrS9n[R"U5n[R"XXUS9n [R"XxUS9n X4$) z See function `_kneighbors_reduce_func `_. zk must be an integer not .rN)axiskthr) isinstancenpndarraysize RuntimeErrorintlenshapearange argpartitionargsortsortfliptake) Xkrlargest sample_rangesorted_indicessorted_distances sorted_valuesarktopk_sorted_indicestopk_sorted_valuess T/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/reference/ops/op_topk.pytopk_sorted_implementationr$ s !RZZ  66Q;!:1%qAB B aD AA 177|qTQYyy,QW5 a<__Qq1uEN+ArrE2N+bjj+G)HIIN __aRdAFN+ArrE2N+bjj!.,H*I)IJJN>9://ZZ-NGGA)M; 9 ))A,C''.DA$?  22c\rSrSrSSjrSrg) _CommonTopK5cUSnUS: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)rrr$astyper int64)selfdatainkrrrrsortis r# _common_run_CommonTopK._common_run6sK Fqyttc$**o'=0$H ll288,--r%N)r)__name__ __module__ __qualname____firstlineno__r1__static_attributes__r3r%r#r'r'5s.r%r'c\rSrSrSSjrSrg)TopK_1INc.[RXU/US9$r*r r'r1)r-r.rrs r#_run TopK_1._runJs&&tA3T&BBr%r3)NNr4r5r6r7r?r8r3r%r#r:r:I Cr%r:c\rSrSrSSjrSrg)TopK_10ZNc*[RXX#S9$r=r>)r-r.r/rs r#r? TopK_10._run[s&&t3&BBr%r3)NrAr3r%r#rDrDZrBr%rDc\rSrSrSSjrSrg)TopK_11kNcNUS;a [S5e[RXX#US9$)r*)Trz.TopK does not implement anything for sorted=0.)rr)rr'r1)r-r.r/rrsorteds r#r? TopK_11._runls0  "OP P&&t37&SSr%r3)NNNrAr3r%r#rIrIksTr%rI) numpyr onnx.reference.op_runrr$r'r:rDrIr3r%r#rPsI ''3T.%.(C[C"CkC"TkTr%