o 2hp@sddlmmZddlmZddlmZddlmZd-ddZ d-dd Z d.d d Z d dZ d-ddZ d/ddZddZddZddZd0ddZd0ddZ d1dd Z   d2d!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd3d+d,ZdS)4N)Type)OpenVINOKerasTensor) get_ov_outputFcCtd)Nz4`segment_sum` is not supported with openvino backendNotImplementedErrordata segment_ids num_segmentssortedr Z/var/www/html/chatgem/venv/lib/python3.10/site-packages/keras/src/backend/openvino/math.py segment_sumrcCr)Nz4`segment_max` is not supported with openvino backendrrr r r segment_maxrrTc Cs^t|}tj|tjd}d}|rdnd}t|||d|}|d}|d}t|t|fS)N)dtypevaluenonemaxr)rov_opsetconstantri32topkoutputr) xkr k_tensoraxis sort_type topk_nodevaluesindicesr r rtop_ks   r%cCr)Nz1`in_top_k` is not supported with openvino backendr)targets predictionsrr r rin_top_krr(c Cs(t|}|durtdgtjd}t||dd}d}t|tr(t |}t|tjd}td| d}t ||dd}t |d}t |||d}t||d}t|d} t| ||d} t| d} |st||d}t|| d} t| S)NrrFT)rrrrrrreshape isinstancetuplelistget_element_type reduce_max is_finiteselectsubtractexp reduce_sumlogsqueezeaddr) rr keepdims flatten_shape const_zeror.r/norm_max norm_max_sub exp_norm_maxsum_exp log_sum_expr r r logsumexp%s( r?reducedcCr)Nz+`qr` is not supported with openvino backendr)rmoder r rqr>rBcCr)Nz:`extract_sequences` is not supported with openvino backendr)rsequence_lengthsequence_strider r rextract_sequencesBrrFcCr)Nz,`fft` is not supported with openvino backendrrr r rfftHrCrHcCr)Nz-`fft2` is not supported with openvino backendrrGr r rfft2LrCrIcCr)Nz-`rfft` is not supported with openvino backendrr fft_lengthr r rrfftPrCrLcCr)Nz.`irfft` is not supported with openvino backendrrJr r rirfftTrCrMhanncCr)Nz-`stft` is not supported with openvino backendr)rrDrErKwindowcenterr r rstftXsrQcCr)Nz.`istft` is not supported with openvino backendr)rrDrErKlengthrOrPr r ristft^s rScCsDt|}td|d}t|d}tt||dS)Nrr)rrrr-rsqrtrdivide)r const_onerTr r rrsqrtjsrWcCs t|}t|d}t|S)Nr)rrerfrr)rrXr r rrXqsrXcCr)Nz/`erfinv` is not supported with openvino backendrrGr r rerfinvwrCrYcCr)Nz.`solve` is not supported with openvino backendr)abr r rsolve{rCr\cCr)Nz-`norm` is not supported with openvino backendr)rordr r7r r rnormrCr^)NF)T)r@)N)rNT)NrNT)NNF)openvino.runtime.opset14runtimeopset14ropenvinorkeras.src.backend.openvino.corerrrrr%r(r?rBrFrHrIrLrMrQrSrWrXrYr\r^r r r rs4