ij4&ddlZddlmZddlZddlmZddl m Z ddl m Z ddl m Z ddl m Z ddlmZdZd Zd Zd Zd#d Zd#d Zd#dZd#dZd$dZdZd#dZdZdZd%dZdZdZdZ d&dZ!d&dZ" d'dZ#dZ$ d(dZ%dZ&dZ'd Z(d!Z)d"Z*y))N)Type)OpenVINOKerasTensor)cast) get_ov_output)standardize_dtype)stackc|tj|tjdgtjdj d}tj |tjd|jj dStj||jj dS)NrF keep_dims)ov_opset reduce_maxconstantri32outputaddget_element_type) segment_ids num_segmentsmax_ids t/var/www/html/emotional.easysim.app/public_html/venv/lib/python3.12/site-packages/keras/src/backend/openvino/math.py_resolve_num_segmentsrs$$ **A39U &) || H%%a)@)@)BC &)    k224  fQic tj|tjd|jj d}tj |||j d}tj |tjdtjj d}tj|tjd|jj d}|j} | jjr| jjnd} | dkDrYtj|tjj d} tj| tjdgtjtjdgtjtjdgtjtjdgtjj d} tj |tjdtjj d} tj | | gdj d}nGtj |tjdtjj d}tj||jj d}tj"||j d}tj$||||j d}tjdgtjj d}tj |tjdtjj d}tj|||tjdgtjj dtjdgtjj dj dS) zEScatter-based segment reduction used by segment_sum and segment_prod.rr  output_typer startstopstepaxesaxis) reduction)rlessrrrselect unsqueezerrrget_partial_shaperank is_static get_lengthshape_ofsliceconcat broadcastscatter_nd_update)datarrr&init_val is_negativesafe_segment_idsindicesnum_segments_plus_1 data_shaper+data_shape_node rest_shape num_seg_node buffer_shape init_val_nodebuffer scatteredr ends r_segment_reduce_via_scatterrBsX--X&&q+*F*F*HI fQi \; fQi  (++B9 fQi #,,h''<+H+H+JK fQi'')J+5??+D+D:?? % % '"D ax"++DdhhGNN  ^^ ##QC2""J<:""A31""A31  &)  )) !2!21dhh!? &)   : &Q &)  )) !2!21dhh!? &) %%h0E0E0GHOO M   | < C CA FF** fQi   qc488 , 3 3A 6E   h''4884  fQi >>  1#txx(//21#txx(//2    fQi rctjdtjj d}tjdgtjj d}tjdgtjj d}tj |tjd|j j d}tj|||j d}tjtj|tj|j d}tj||ddd} | j d} | j d} tj|| |j d} tj|tjd|j j d} tj| | | dj d}tj||j d}tj|||||j dS) zCSort + segment_max reduction shared by segment_max and segment_min.rr rminvalue)r%modesortLOWEST) fill_mode)rrrrrr'rr(squeezer.topkgatherr segment_maxr)r/)r3rrzerozero_1done_1dis_negsafe_segnrK sorted_segsort_idx sorted_data nseg_plus1resultnseg_1ds r_segment_reduce_via_ov_maxrZ\s   Q ) 0 0 3DTXX.55a8G   sDHH - 4 4Q 7F ]]X&&q+*F*F*HI fQi v|[AHHKH (97  fQi ==115w GDQJ{{1~H//$$7>>qAKh''<+H+H+JK fQi ! !Zx fQi   t4;;A>G >>&'7FG D K KA NNrc|t|}t|}t||}t|||dd}t|S)Nsumrr&r4rrrBrr3rrsortedrXs r segment_sumra{sD  D ,K(lCL ( k<51F v &&rcvt|}t|}t||}t|||}t|SN)rrrZrr_s rrMrMs;  D ,K(lCL 'k< HF v &&rct|}t|}t||}tj|j d}tjt |||j d}t |SNr)rrrnegativerrZr)r3rrr`neg_datarXs r segment_minrhsr  D ,K(lCL  &--a0H   "8[,G fQi  v &&rc|t|}t|}t||}t|||dd}t|S)Nprodr r]r^r_s r segment_prodrksD  D ,K(lCL ( k<6AF v &&rct|}tj|tj}d}|rdnd}tj |||d|}|j d}|j d}t|t|fS)NdtyperrEnonemaxrr )rrrrrrKrr) xkr`k_tensorr% sort_type topk_nodevaluesr7s rtop_krws|aA  $((3H D!vI a4 BI   a Fq!G v &(;G(D DDrcddlm}tjdtj }tj t||jd}t|}t||d}tj|dz gtj jd}tjdtj jd}tj|||jd}tj||jd}|||d} tj| |jd} tj| |jd} t| S)Nr)take_along_axisr rmrr$) keras.src.backend.openvino.numpyryrrrrr)rrrwrLrJ greater_equalr) targets predictionsrrry one_constant topk_values k_minus_1_idxtopk_values_axistopk_mintargets_valuesmasks rin_top_krs9@$$Q1L  w!7FMMaPG ,K Q'*K%%q1ugTXX>EEaHM(($((;BB1E]$4 fQi ,7>>qAH%['CN%%nlCJJ1MN  ! !.( ; B B1 ED t $$rct|}|\tjdgtjj d}tj ||dj d}d}t|tr t|}tj|tjj d}tjd|jj d}tj||dj d}tj|j d}tj|||j d}tj||j d}tj|j d} tj | ||j d} tj"| j d} |s%tj$||j d}tj&|| j d} t)| S)NrrFT)rrrrrrreshape isinstancetuplelistrr is_finiter(subtractexp reduce_sumlogrJrr) rqr%keepdims flatten_shape const_zerorrnorm_max norm_max_sub exp_norm_maxsum_exp log_sum_exps r logsumexprsaA | ))2$9@@C   Q u 5 < = 2rz.Last dimension of inputs to `cdist` must matchrF)rr*r+r,r- ValueErrorrrrrrr)rmultiplyrsqrtr) rqyx_shapey_shapex_lasty_lastneg2neg3diffsq last_axissummedrXs rcdistrsaAaA!!#G!!#G||',,"9"9";a"?=>>||',,"9"9";a"?=>> R[F R[F F,,    &"3"3"5 5MN N   R * 1 1! 4D   R * 1 1! 4D1d#**1-A1d#**1-A   Q " ) )! ,D   4 & - -a 0B!!"dhh/66q9I  Y 6 = =a @F ]]6 " ) )! ,F v &&rc  t|}|j}t|}tjdgt j }tj|t j jd}tj|dz gt j }tjdt j }tj|||jd} tj| |jd} tjdgt j jd} tj| | gdjd} tj|| djd} tj|t j jd}tj|t j jd}tj| |jd}tjtj||jdtjdt j jdjd}tj ||jd}tj"tjdt j jd||t j jd}tj$|tjdgt j jd}tjt'j(|t&j*jd}tj$||jd}tj||jd}tj| |tjdt j jd}tj,||tj|dz gt j tjdgt j |jd}tj$||jd}tj|gt j jd}tj|||gdjd}tj||djd}t/|S) Nrrr rr$Frmr)r partial_shapelenrrrrr.rrLrJr0rrrdividerranger)nparangeint32r/r)rqsequence_lengthsequence_striderndim zero_const_1d shape_tensorlast_idxaxis0 signal_len_1dsignal_len_scalar minus_oneshape_2dx_2d seq_len_cstride_crnum_seq_scalarrow_stoprow_idx row_idx_2dcol_idx col_idx_2dr7gathered batch_shape num_seq_1d seq_len_1d out_shaperXs rextract_sequencesrsaAooG wsC! I"+y"8adI adB"+y"8adI adB   b8,,bT488<  fQi    b8,,bT488<  fQi ??D$<4;;A>L qtzz?Q D   +,,dhh  fQi "x||B  d # * *1 -F!!!TXX.0A0A"dhh0O fQi !!!TXX.0A0A"dhh0O fQi I##Hdhh7>>qA##Hdhh7>>qA x (*=h*G GG# -s9 K9c t|dgdS)NrFrrrrqs rfftris e 44rc"t|ddgdS)NrrFrrrs rfft2rns R% 88rc"t|ddgdS)NrrTrrrs rifft2rss R$ 77rc<|j}|jdk(r t|dn|}t|}t|j}t j |dz gtjjd}| t j |gtjjd}|jd}t|tr||krdg|z}dg|z} ||z | d<t j |tjjd} t j | tjjd} t j|| | djd}t j|||jd} n%t j||jd} t j| t j dtjt j dtjjd} t j| t j dtjt j dtjjd}|dk(rft j| tj jd} t j|tj jd}t#| t#|fS)Nrrr rrr)rnrrrrrrrrrrintpadrdftrLrrr)rq fft_lengthrx_noder+r# signal_sizelast_dim pad_beginpad_endpad_begin_node pad_end_noderrrs rrfftrxsXIgg2Q A 1 F v|| D   dQhZ 2 9 9! d IcDjG$x/GBK%..y$((CJJ1MN#,,Wdhh?FFqIL\\ jfQi }}VT;7>>qA}}VT*11!4 h488,h.?.?DHH.M fQi  h488,h.?.?DHH.M fQi I##Hdhh7>>qA##Hdhh7>>qA x (*=h*G GGrc|dj}|dk(rt|ddt|ddf}tt|d}t |j }t j|dz gtjjd}|[t j|gtjjd}t j|||jd}n%t j||jd}|dk(r3t j|tjjd}t|S)Nrrrr rr$r)rnrrrrrrrrrrirdftrrr)rqrrrr+r#rrs rirfftrs! II !A$ "D1y$9 : qr!23L |!! "D   dQhZ 2 9 9! rrzJInvalid input type. Expected `float32` or `float64`. Received: input type=zT`fft_length` must equal or larger than `sequence_length`. Received: sequence_length=z , fft_length=>hannhamming]If a string is passed to `window`, it must be one of `"hann"`, `"hamming"`. Received: window=rrrFreflectr PThe shape of `window` must be equal to [sequence_length].Received: window shape=)transpose_framesr)rrn TypeErrorrrstrrrrrrrrrrrrrrf32scipysignal get_windowronesstftrLrr)rqrrrwindowcenterr ori_shapenum_dimsrrrrrl_padr_pad element_typewinwin_nodeframe_size_nodeframe_step_node stft_noderr target_shapetarget_shape_nodes rr r s!)?? $$%GG9 .  O# ))8(9:$ '  &# , ,;;A(D  IaAI9~H!| ))2y}*=txxHOO     Q u 5 < Q #))B-?"B**-))6 ffSE5>*+$$S,7>>qAgg.68$$S,7>>qA'' DHH=DDQGO''BII!LO  8_o fQi!TXX&"dhh' fQi  !TXX&"dhh' fQi  !|IcrN+r:?Q3F.GG $--lDHHELLQO##H.?GNN  ##H.?GNN  I##Hdhh7>>qA##Hdhh7>>qA x (*=h*G GGrc d|dz |zz}||z}||z }tjdtjj d}tj |tjj d}tj |tjdtj|j d}tj |tjdtj|j d} tjtjtj|tjtj| tjdtjtj|tjj d} |dkDrtjgdtjj d} tjdd|gtjj d} tj|| | dj d}tjtj|tjdtjj dtj| tjdtjj dtj||gtjj dgdj d} tj|| dj d}tj|tjgdtjj d}tjgdtjj d}tjtjddgtjj dtj| tjdtjj dtjdgtjj dgdj d}tj|||dj d}tjtjtjd tj| tjdtjj d}tjtj|d z|ztj| j d}tjtj|tjdtjj dtj|tjdtjj dgdj d}tj||dj d}tj|tj||ztjj d}tj|tjdgtjj dtj|tjdtjj dtjdgtjj dtjdgtjj dj d}tj|tj|tjj d}tjtj|tjdtjj dtj|gtjj dtj|tjdtjj dgdj d}tj||dj d}tj |tjdgtjj dd j d}tj|tjdgtjj dtj| tjdtjj dtjdgtjj dtjdgtjj dj d}|| fS) aBPerform overlap-and-add using OpenVINO ops. Takes a tensor x of shape [batch, num_sequences, fft_length] and reconstructs a time-domain signal by striding each frame by `sequence_stride` and summing the overlapping contributions. Returns the reconstructed signal and its length as an OV scalar node. r rr)rrrrF)rrr )rrrrrr )rrrrrr.rLrrrrr0r)r transposer/r)rqrrnstep padded_len pad_amountrNr flat_batch num_sequences output_sizerr overlap_shape pad_begin_n pad_end_noverlapping_dim_size total_innerr slice_len inner_size sum_shapes r_overlap_sequences_ovr*s aO3 3E(Jj(J   Q ) 0 0 3Dtxx8??BG""1dhh/ fQiOO""1dhh/ fQi ,,   otxx 8   mX->->q$((-K L  *dhh/   fQi A~%%i:AA!D ##Q:$6AHHK LLIw ; B B1 EOO   H--a: fQi   x00DHH= fQi   uo6 A H H K    fQi M5188;A 1h// dhhGHOO  A ##L$((;BB1EK   q!fdhh / 6 6q 9   x00DHH= fQi   qc488 , 3 3A 6     fQi  Q Y ;BB1EA$,,(++Atxx8-H!TXX& fQi##%!)o5txx@ fQiOO   H--a: fQi   X..q$((; fQi     fQi M5188;A!!%/1488< fQi  1#txx(//29h&7&7488&DELLQO1#txx(//21#txx(//2    fQi ""h//J fQi   H--a: fQi   ugtxx 0 7 7 :   H--a: fQi    fQi Iu-44Q7A 1#txx(//2  fQi   1#txx(//2;(9(9!TXX(FGNN   1#txx(//21#txx(//2  fQi k>rc Pt|tr|dvrwtd||gtj|tj }t |jdk7s|jd|k7rtd|j|dj}|t||} t| } | j} | tjk(r3tj| tj j#d} | j%} | j&j)} | D cgc] } | j*rdn| j)"}} | dk(rItj,| tj.dtj0j#d} n| dkDrtj2| tj0 j#d}tj4|tj.| dz tj0tj.dtj0j#d}tj6tj.dgtj0j#dtj,|tj.dtj0j#dtj.|gtj0j#dgdj#d}tj8| |d j#d} t;| ||\} }|r|dznd}|tj<| tj.|gtj0j#dtj.||zgtj0j#dtj.dgtj0j#dtj.dgtj0j#dj#d} n|dkDrtj<| tj.|gtj0j#dtj.t>gtj0j#dtj.dgtj0j#dtj.dgtj0j#dj#d} |rtj4tj2| tj0 j#dtj.dtj0tj.dtj0j#d}tj@|tj.|dztj0j#d}tj<| tj.dgtj0j#dtj,|tj.dtj0j#dtj.dgtj0j#dtj.dgtj0j#dj#d} | dk(rItjB| tj.dgtj0j#d} ny| dkDrt|dd }|Dcgc]}||nd c}dgz}tj8| tj.|tj0j#dd j#d} |d k(r3tj| tjj#d} tE| St|tr!tFjHjK||}n%tj|tj }t|d}t|d}|j} | tjk(rftj|tj j#d}tj|tj j#d}tj,|tj.dtj0j#d}tj,|tj.dtj0j#d}tj6||gd j#d}|j%j&j)}tMtO|dz |dz |dz |dz gz}tjP|tj.|tj0j#d} |dz }!|!dkDrjtj2| tj0 j#d}"tj<|"tj.|!gtj0tj.t>gtj0tj.dgtj0tj.dgtj0j#d}#tj8| tj6tj.dgtj0j#d|#gd j#dd j#d} tj.|jStjTtj j#d}$tj.|tj0j#d}%tj.|tj0j#d}&|3tj.|tj0j#dnd}'tjV| |$|%|&|d |'j#d}(|!dkDr|j%} | D cgc] } | j*rdn| j)"c} dd })|)Dcgc]}||nd c}dgz}tj8|(tj.|tj0j#dd j#d}(|d k(r3tj|(tjj#d}(tE|(Scc} wcc}wcc} wcc}w)N>rrrrmr rrrrrFrrr$r)r  normalized signal_length),rrrrasarrayrrrrnrrrrrrrrrr*r+r- is_dynamicr)rrr.rLr0rr*r/ INT32_MAXrrJrrrrrrrastyperistft)*rqrrrlengthr r rrframesrori_partial_shaperdimori_shape_list frames_shp num_seq_node flatten_shp_ start_valcur_lenend_node batch_dimsdrx_realx_imagreal_expimag_expstackedr+permr3num_batch_dimsr:trailingrrrsignal_length_noderXrs* rr2r2s &# , ,;;A(D   jjrzz2 syy>Q #))B-?"B**-))6  ! I ~q*%v&..0 488 #%%fdhh7>>qAF"446$))446) NND(8 8  q=''))!TXX6fQi \!**6txxHOOJ$??!!(Q,9!!!TXX.fQi  #//%%rdDHH5<%%qc4884;;A>  &) "//%%f$((CJJ1M%%a2%%a2&)  $,,%%jAotxx@&)"%%qc4884;;A>&& ("3"3Atxx"@fQi%%qc4884;;A>%%qc4884;;A>&) q=%%))1#txx8fQi \',J@JK1AB6KOL%%!!,9@@CfQi   !%%fdhh7>>qAF"6**&#ll%%fo>jjrzz2 1Q4 F 1Q4 F**,Ltxx!!&$((3::1=!!&$((3::1=!!!!"dhh/ fQi !!!!"dhh/ fQi oox2<CCAFG  $ $ & + + 6 6 8D dQh D1HdQhq#A AD   ""42  fQi AXN"++DdhhGNN  >>    ~. 9   yk488 4   qc488 ,   qc488 ,  &)   OO""B4299!EEaHG   b" % , ,Q /E||H--e4;;A>?FFqIH   *E299!<  fQi     HOOGU + 2 21 5  fQi r ""rc$ddlm}||dS)Nr)slogdetr )rzre)rqres rlogdetrfs8 1:a=r)NF)T)Frc)rT)NrT)+numpyropenvino.opset16opset16r scipy.signalropenvinorkeras.src.backend.openvino.corerrrrrzrr0rrBrZrarMrhrkrwrrrrrrrrrrr r*r2rLrNrQrcrfrrrns#?09=2   =@O>'' ''E%8,2'4>'B(HV5 9 8 %HP&,LPcHLW~   T'nK$ L1#hr