=Vh&: dZddlZddlmZddlmZddlmZddlm Z ddl m Z ddlmZdd l mZdd l mZdd l mZdd lmZdd lmZddlmZddlmZm Z m!Z!ddl"m#Z#edZ$ejJejLedd!de#e!e$fde#e!e$ffdZ'edejPe'Z)e'jTjVZ,de#e!e$fde#e!e$ffdZ-ejJejLedd!de#e e!ej\ffdZ/edejPe/Z0e/jTjVZ1de#e e!ej\ffdZ2ejJejLedd!dZ3edejPe3Z4e3jTjVZ5d Z6y)"zUPython wrappers around TensorFlow ops. This file is MACHINE GENERATED! Do not edit. N) pywrap_tfe)context)core)execute)dtypes)annotation_types)op_def_registry)ops)op_def_library)deprecated_endpoints)dispatch) tf_export)TypeVarListAny) Annotated) TV_XlaClusterOutput_Tz_atypes.BFloat16z _atypes.Boolz_atypes.Complex128z_atypes.Complex64z_atypes.Float16z_atypes.Float32z_atypes.Float64z_atypes.Float8e4m3b11fnuzz_atypes.Float8e4m3fnz_atypes.Float8e4m3fnuzz_atypes.Float8e5m2z_atypes.Float8e5m2fnuzz _atypes.Halfz _atypes.Int16z _atypes.Int32z _atypes.Int4z _atypes.Int64z _atypes.Int8z_atypes.QInt16z_atypes.QInt32z _atypes.QInt8z_atypes.QUInt16z_atypes.QUInt8z_atypes.Resourcez_atypes.Stringz_atypes.UInt16z_atypes.UInt32z _atypes.UInt4z_atypes.UInt64z _atypes.UInt8z_atypes.Variantxla_cluster_outputinputreturnc >tjxstj}|j}|jr t j |d||}|St||fd}|tur|S t/j0d||\}}}}|dd}t3j4r7d|j7df} |j8} t3j:d| | ||\}|S#t j$r }tj||Yd}~nd}~wt j$rYnwxYw t||fd}|tur|St|||S#t j$rYtt f$rHt#j$t&dt)||}|t"j*j,ur|cYSwxYw#tt f$rHt#j$t&dt)||}|t"j*j,ur|cYSwxYw)zOperator that connects the output of an XLA computation to other consumer graph nodes. Args: input: A `Tensor`. name: A name for the operation (optional). Returns: A `Tensor`. Has the same type as `input`. XlaClusterOutputN)namectx)rrT)_contextr_thread_local_datais_eagerrTFE_Py_FastPathExecute_core_NotOkStatusException_opsraise_from_not_ok_status_FallbackException"_dispatcher_for_xla_cluster_outputNotImplemented!xla_cluster_output_eager_fallback_SymbolicException TypeError ValueError _dispatchr rdict OpDispatcher NOT_SUPPORTED_op_def_library_apply_op_helper_executemust_record_gradient_get_attr_typeinputsrecord_gradient) rr_ctxtld_resulte__op_outputs_attrs _inputs_flats S/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/compiler/jit/ops/xla_ops.pyrrs    0h..0$ #\\ 11  $/g n,1 Gn$ n  )::%d4Aq#x QK' ""$3%%c* +F::L L&'; (' .W  & &- ##At,,  # #    2 $.$ g  & . d&&  # #  z " "" Du4$@ g  ..<< <  Z    b$U"> Gi,,::: n  sPC&3G&D-9DD-,D-1E EG,AGGAHHzraw_ops.XlaClusterOutputctj|g|g\}\}|g}d|f}tjdd||||}tjrtjd||||\}|S)NrsXlaClusterOutputr5attrsrrr)r2args_to_matching_eagerrr3r6)rrr_attr_Tr?r>r9s r@r(r(`s|55ugsBG'8E, >&   0!L#)s ?' ""$ L&'; (' . xla_launch resourcesc tjxstj}|j}|jr! t j |d||||d|d| }|St||||||fd}|tur|St/|t0t2fstd|zt5|} t/|t0t2fstd |z|D cgc]} t7j8| d}} t;j<d||||||\} } } }|dd}|s| St7j>r{d | jAd d | jAd d | jCd d| jAdd| jAdf }| jD}t7jFd||||S#t j$r } tj| |Yd} ~ nd} ~ wt j$rYnwxYw t||||||fd}|tur|St|||||||S#t j$rYtt f$rLt#j$t&dt)||||||}|t"j*j,ur|cYSwxYwcc} w#tt f$rLt#j$t&dt)||||||}|t"j*j,ur|cYSwxYw) aXLA Launch Op. For use by the XLA JIT only. Args: constants: A list of `Tensor` objects. args: A list of `Tensor` objects. resources: A list of `Tensor` objects with type `resource`. Tresults: A list of `tf.DTypes`. function: A function decorated with @Defun. name: A name for the operation (optional). Returns: A list of `Tensor` objects of type `Tresults`. XlaLaunchTresultsfunctionN)rLrMrrr) constantsargsrIrLrMrBExpected list for 'resources' argument to 'xla_launch' Op, not %r.AExpected list for 'Tresults' argument to 'xla_launch' Op, not %r. TconstantsTargs Nresources)$rrrrrr r!r"r#r$r%_dispatcher_for_xla_launchr'xla_launch_eager_fallbackr)r*r+r,r rHr-r.r/ isinstancelisttuplelenr2 make_typer0r1r3get_attr _get_attr_intr5r6)rNrOrIrLrMrr7r8r9r:_attr_Nresources_tr;r<r=r>r?s r@rHrHms+"    0h..0$ #\\ 11 k4D)Z*h(gn2) D)Xx?GGn$ n Ie} -  #%. / 00^ HtUm ,  #%- . //r9s r@rVrVsS Ie} -  #%. / 00^ HtUm ,  #%- . //r{d | jAd d| jAdd| jAdd| jAdd| jAdf }| jB}t5jDd||||S#t j$r } tj| |Yd} ~ nd} ~ wt j$rYnwxYw t||||||fd}|tur|St|||||||S#t j$rYtt f$rLt#j$t&dt)|||||| }|t"j*j,ur|cYSwxYwcc} wcc} wcc} w#tt f$rLt#j$t&dt)|||||| }|t"j*j,ur|cYSwxYw)abXLA Launch Op. For use by the XLA JIT only. Args: args: A list of `Tensor` objects. Tresults: A list of `tf.DTypes`. constants: A list of `ints`. resources: A list of `ints`. function: A function decorated with @Defun. name: A name for the operation (optional). Returns: A list of `Tensor` objects of type `Tresults`. XlaLaunchV2rLrNrIrMN)rLrNrIrMrrr)rOrLrNrIrMrDExpected list for 'Tresults' argument to 'xla_launch_v2' Op, not %r.EExpected list for 'constants' argument to 'xla_launch_v2' Op, not %r.EExpected list for 'resources' argument to 'xla_launch_v2' Op, not %r.rS)#rrrrrr r!r"r#r$r%_dispatcher_for_xla_launch_v2r'xla_launch_v2_eager_fallbackr)r*r+r,r rgr-r.r/rWrXrYr2r[make_intr0r1r3r\r5r6)rOrLrNrIrMrr7r8r9r:r__ir;r<r=r>r?s r@rgrgs"    0h..0$ #\\ 11 mT4X{; :xAgn2, xIx?GGn$ n HtUm ,  &(0 1 22r9s r@rnrnEs~ HtUm ,  &(0 1 22rs  67179FK3IC86%%'N  %% ((  <i-B(BC<S\]`bw]wSx<!)&<|9978HZ9[\%7%Q%Q%Z%Z" Ys4>>*+E F 'AAJJ)DIwO_O_D_:`2 %% (( ?X)&Xt/i-.~t~~m/LM - G G P PrG