AVhr-dZddlZddlmZddlmZddlmZddlmZddlmZddlmZdd l m Z dd l m Z dd l m Z dd l m Z dd l mZddl mZddl mZddl mZddl mZddl mZddl mZddlmZddlmZddlmZddlmZddlmZddlmZej@Z gZ!gZ"dZ#d)dZ$d*dZ%dZ&e$ejNe_'e$ejPe_(e$ejRe_)e$ejTe_*e$ejVe_+e$ejXe_,e$ejZe_-e$ej\e_.e$ej^e_/e$ej`e_0e$ejbe_1e$ejde_2e$ejfe_3e$ejhe_4e$ejje_5e$ejle_6e$ejne_7e$ejpe_8e$ejre_9e$ejte_:e$e jve _;e$ejxe_<e$ejze_=e$ej|e_>e$ej~e_?e$eje_@e$eje_Ae$eje_Be$eje_Ce$eje_De$eje_Ee$eje_Fe$eje_Ge$eje_He$eje_Ie$eje_Je$eje_Ke$eje_Le$eje_Me$eje_Ne$eje_Oe$eje_Pe$eje_Qe$e je _Re$e je _Se$e je _Te$e je _Ue$e je _Ve$eje_We$eje_Xe$eje_Ye$eje_Ze$eje_[e$eje_\e$eje_]e$eje_^e$eje__e$eje_`e$eje_ae$eje_be$eje_ce$eje_de$e je _ee$eje_fe$eje_ge$eje_he$eje_ie$eje_je$eje_ke$eje_le$eje_me$eje_ne$eje_oe$eje_pe$eje_qe$eje_re$eje_se$eje_te$eje_ue$eje_ve$eje_we$eje_xe$eje_ye$eje_ze$eje_{e$eje_|e$eje_}e$eje_~e$eje_e$eje_e$eje_e$eje_e$eje_e$eje_e$ej e_e$e j e _e$e je _e$e je _e$e je _e$e je _e$e je _e$e je _e$e je _e$e je _e$e je _e$e j e _e$e j"e _e$e j$e _e$e j&e _e$e j(e _e$e j*e _e$e j,e _e$e j.e _e$e j0e _e$ejNe_'e$ej2e_e$ejXe_,e$ej4e_e$ej6e_e$ej8e_e$ej:e_e$ej<e_e$ej>e_e$ej@e_e$ejje_5e$ej`e_0e$ejBe_e$eje_Ae$eje_Be$ejDe_e$ejne_7e$ejFe_e$eje_Fe$ejHe_e$ejJe_e$ejte_:e$eje_He$ejLe_e$ejNe_e$eje_Ke$eje_Me$eje_Ne$ejle_6e$eje_Pe$eje_Qe$ejPe_e$ejRe_e$ejTe_e$eje_Ie$ejVe_e$ejXe_e$eje_Oe$ejZe_e$ej\e_e$ej^e_e$ej e_e$ej`e_e$ejbe_e$ejde_e$ejfe_e$ejhe_e$ejje_e$ejle_e$ejne_e$ejpe_e$ejre_e$ejte_e$ejve_e$ejxe_e$ejze_e$ej|e_e$eje_e$ej~e_e$eje_e$eje_e$eje_e$eje_e$ejVe_+e$eje_e$eje_e$eje_e$eje_e$eje_Te$eje_e$eje_e$ejTe_*e$ej e_e$eje_e$eje_e$ejZe_-e$ej&e_e$eje_e$eje_e$eje_e$ej.e_e$eje_e$eje_e$eje_e$eje_Re%eje_e%eje_e%eje_e%eje_e%eje_e%eje_e$ejde_e%eje_e%eje_e%eje_e%eje_e%eje_e%eje_e%eje_e%eje_e%eje_e%ejd e_e%ejd e_e%ejd e_e%ejd e_e%ejd e_e%e jd! e _e%e jd! e _e%e jd! e _e$eje_e%ejd"d#$e_eje_eje_eje_eje_eje_ejej_ejej_ejNej_gd%ZejjD]0ZeevreejeZeejee2ejejeejd&ejdeejd'ejeejd(ejy)+z!Support for WeakTensor in TF ops.N)composite_tensor) constant_op)flexible_dtypes)ops)tensor) weak_tensor) array_ops)clip_ops) gen_array_ops)gen_bitwise_ops) gen_math_ops) image_ops)math_ops)nn_impl)nn_ops)resource_variable_ops)special_math_ops) np_array_ops) np_math_ops) tf_logging)core)dispatch) tf_decoratorct|tjr|j}t|tj st|t jrtj|||Stj|||S)z$Converts/casts the input x to dtype.)dtypename) isinstancer WeakTensor to_tensorrTensorrCompositeTensorrcastrconvert_to_tensor)xrrs U/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/ops/weak_tensor_ops.py_convert_or_castr&6sl ;))* A4;;:  ) )$ ==%d 33  %d ;;ctj.tjj }t |fd}t j|}t|tj||S)aInfers input type and adds WeakTensor support to unary ops. This wrapper infers input type according to the auto dtype conversion semantics - Tensor and NumPy inputs as Tensor of corresponding dtype and WeakTensor and python inputs as WeakTensor of corresponding dtype. If the inferred input dtype is "weak" and the op doesn't specify a return dtype, returns WeakTensor. c :tjs|i|Sj|i|}|j|j}| }t |t js|jdddi|S tj|\}}t#||d| <t%j&di||S#t$rGtjdjdjdt!|ddi|cYSwxYw)Nr'The new dtype semantics do not support .(!). Falling back to old semantics.r$)r is_auto_dtype_conversion_enabledbindapply_defaults argumentsrrr getr result_typeNotImplementedErrorloggingwarning __module____name__typer&r convert_to_weak_tensor_or_tensor) argskwargsbound_arguments bound_kwargsr$ target_typeis_weakop signature x_arg_names r%wrapperz-weak_tensor_unary_op_wrapper..wrapperVs.  / / 1    $innd5f5O""$",,LZ A 1fmm$   GT * 6  ,   ,88;k7 0;DL  7 7 \G    oo mm_Abkk]!DG95   ,   sC A DD) inspectrCiter parameterskeysnextrmake_decorator0_update_weak_tensor_patched_ops_in_dispatch_dict_TF_UNARY_APISappend)rBrD arg_namesrErCs`` @r%weak_tensor_unary_op_wrapperrPHsu#)Y))..01IiJ>  ' 'G 4'37; .r'c.tjtjj }t | t |fd}t j|}t|tj||S)a(Determines result promotion type and adds WeakTensor support to binary ops. This wrapper first infers dtype of any Tensor, WeakTensor, python/numpy inputs. Then, both inputs are promoted to the correct promotion result dtype. If the result promotion dtype is "weak", returns WeakTensor. ctjs|i|S j|i|}|j|j}| }| } t j ||\}} dk(rp||jk7rItdjd|jd |d |d |jd |d jdt||d| <di|S dk(rot!|t"j$r|j'| <|:d}||k7rdi|St!|t(j*r7t||d| <n&||d<n t||d| <t||d| < dk(rd}t#j,di||S#t$rStjdjdjdt|dt|d di|cYSwxYw)Nr*r+r,z, r-variable_methodz%Variable dtype is immutable. Calling z of Variable (with dtype z) on z requires converting z to zF. This is disabled in the current promotion semantics. Please convert z manually before calling yconstantFr$rcomparison-methodr.)rr/r0r1r2rr4r5r6r7r8r9r:r TypeErrorr&rrrrrr r;) r<r=r>r?r$rTr@rArBrCspecial_handlingrD y_arg_names r%rEz.weak_tensor_binary_op_wrapper..wrappersS  / / 1    $innd5f5O""$",,LZ AZ A  ,88A>k7,,  ? }M33477)5D&&'SQWWI6GGHcJ335;;-q BC C "2![#!Fl:  ,  Z ' A{-- .#$;;= Z   ! #l# # a %%5ac%J,z " + W!1![#!Fl:!1![#!Fl:..g  7 7 \G a   oo mm_Abkk]!DG9BtAwi@   ,   sFAG87G8) rFrCrGrHrIrJrrKrL_TF_BINARY_APISrN)rBrYrXrOrErCrDs``` @@r%weak_tensor_binary_op_wrapperr[s#)9'',,./)I*iJ??B  ' 'G 4'37;! .r'cVtj}|j}||vr ||||<yy)aUpdate dispatch dictionary to store WeakTensor patched op references. _TYPE_BASED_DISPATCH_SIGNATURES in dispatch.py stores mappings from op reference to all the dispatchers it's registered with. We need to update this dictionary to add a mapping from the patched-op reference to the signature dictionary the unpatched-op reference is mapped to. This ensures that dispatch can be reigstered and unregistered with monkey-patched ops. N)r_TYPE_BASED_DISPATCH_SIGNATURES __wrapped__) patched_op dispatch_dict unpatched_apis r%rLrLs6::-((-m# -m r>s1(837+.3+*/1.+*)(7287<(+/)99<$4nUr =&,HLL9 01B1BC,X]]; ,X]]; ,X]]; -hnn=-hnn=/0@0@A3H4H4HI,X]]; 9 /x?-hnn=/0@0@A,X]]; ,X]]; +HLL9 -hnn=,X]]; -hnn=5o6L6LM1,2D2DE 01B1BC 1,2D2DE 01B1BC 1,2D2DE / 0@0@A 01B1BC 3L4H4HI / 0@0@A 01B1BC 1,2D2DE 2<3F3FG / 0@0@A 1,2D2DE / 0@0@A 6|7N7NO 01B1BC / 0@0@A 01B1BC 2<3F3FG / 0@0@A 01B1BC 3I4H4HI 6y7N7NO 293F3FG 5i6L6LM ;    ,FLL9 01B1BC*6;;7 1&2D2DE4V5J5JK,W]]; )&** 5 *6;;7 *6;;7 .v?:  )E ))) %;  6i6L6LM 5h6L6LM56F6L6LM67G7N7NO;    <   77G7N7NO9;:;:;:;:::: "> """9 383F3FG5h6L6LM3H4H4HI6x7N7NO283F3FG5h6L6LM3H4H4HI6x7N7NO283F3FG283F3FG5h6L6LM7 .hnn=01B1BC 7  ;  5Y5J5JK 7  #? ### &B &&& "2)2D2DE 4Y5J5JK 9  :  8  ;  11B1BC 3I4H4HI 6y7N7NO 9  393F3FG 5i6L6LM /{? 3K4H4HI 01B1BC 1+2D2DE 1+2D2DE 2;3F3FG 1+2D2DE 2;3F3FG 6{7N7NO / 0@0@A / 0@0@A / 0@0@A 4[5J5JK .{? / 0@0@A 2;3F3FG .{? / 0@0@A 01B1BC / 0@0@A .{? 01B1BC .{? 3K4H4HI 2;3F3FG 5k6L6LM .{? / 0@0@A / 0@0@A .{? / 0@0@A 2;3F3FG / 0@0@A 01B1BC 01B1BC 3K4H4HI / 0@0@A 1+2D2DE / 0@0@A / 0@0@A 2;3F3FG 01B1BC 01B1BC 01B1BC 2<3F3FG 2<3F3FG 1,2D2DE 6|7N7NO 3L4H4HI !=""" 11B1BC 3L4H4HI 2<3F3FG 01B1BC 4\5J5JK 4\5J5JK 6|7N7NO 7 4L4H4HI 01B1BC 2<3F3FG 2<3F3FG 5l6L6LM 01B1BC / 0@0@A 01B1BC / 0@0@A 4\5J5JK 5l6L6LM 01B1BC 1,2D2DE 01B1BC 3L4H4HI 2<3F3FG 1,2D2DE 1,2D2DE 3L4H4HI / 0@0@A / 0@0@A 4\5J5JK 5l6L6LM 01B1BC 2<3F3FG / 0@0@A 6|7N7NO -X\\: 01A1AB 1(2C2CD8 0@2;3E3EF 283F3FL/@3H4G4GH9 6l6K6KL 9 7|7M7MN 9 7|7M7MN 1,2C2CD 4+> 5+> 3)< 1 ':1 ':8.?<2C<2C&B&&&("5Wz   ((#00!++#00 **%-$4$4 !!-!1!1 !) < 449H %%  x0+ + (K8 9" !!+"8"89  \-@-@A  6 8M8MN  ;+=+=>r'