Pi@<ddlTddlTdZGddZdZdS))*c,|j |jjSdSN)origin signaturenodes h/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/einx/tracer/optimize.py get_signaturer s ; {$$tceZdZdZdZdS) Optimizerc"i|_d|_dS)NF)optimized_nodeschanged)selfs r __init__zOptimizer.__init__ s! r c t|jvrjt|St|tr`|jt dt|j|j|j|j |j}nt|trt|j trt|dkrt|j jdkrg|j j}t!|j j|j jj j|gt%|d}d_nt|dkrEt)|j j|j jkr|j j}d_net|dkrEt)|j j|j jkr|j j}d_n t|dkrgt+|j jt+t/t1|jkr|j j}d_nt2jd|j j}fd }t2j|||j jjt|}t!|j j|j j|j j||j jfd |j jD|j j|j j n|}nt|t*rfd |D}nit|t@rtAfd |D}n8t|tBr!fd|"D}n|}|jt|<|S)NzFunction output is None)funcargskwargs virtual_argsoutputreshape)rrT broadcast_to transposec*|Sr)__copy__rs r z$Optimizer.__call__..EsT]]__r cft|jvsJ|jt|<dSr)idr)new_noder rs r storez!Optimizer.__call__..storeHs6#%d88t/C#C#C#C#C9A,RXX666r c~g|]9\}}jt|jt|f:S)rr!).0 tensor_in tensor_outrs r z&Optimizer.__call__..UsS))) !6 :!% 4R ]] C $ 4R ^^ D)))r )rrinplace_updatescomment depend_onc&g|] }|Sr%r%r&xrs r r)z&Optimizer.__call__..bs!...AQ...r c3.K|]}|VdSrr%r.s r z%Optimizer.__call__..ds+33TT!WW333333r c.i|]\}}||Sr%r%)r&kvrs r z&Optimizer.__call__..fs'<<??G%e}}"+    H $(DLL!$''944!$+"4559JJJ $tDK$677H#'DLL!$''>99!$+"4559JJJ $tDK$677H#'DLL"4((K77DAX'>44dk6H(($BBBBBN++E3CT[EWXXX#3BtHH=HT[^,,T[-..T[/00/"&+"7)))) :>9T )))!% 3"&$t{'<"="="  d # # .......HH e $ $ 3333d33333HH d # # <<<rTsv________Dr