AVhG+ dZddlZddlmZddlmZddlmZddlm Z ddl m Z ddlmZdd l mZdd l mZdd l mZdd lmZdd lmZddlmZddlmZm Z m!Z!ddl"m#Z#eddddddZ$d'de#e!e$fde%de&de%de#e!e$ff dZ'edejPe'Z)de#e!e$fde%de&de%de#e!e$ff dZ*eddddddZ+d'de#e!e+fde#e!e+ffd Z,ed!ejPe,Z-de#e!e+fde#e!e+ffd"Z.ed#dddddZ/d'de#e e!e/fde%de#e!e/ffd$Z0ed%ejPe0Z1de#e e!e/fde%de#e!e/ffd&Z2y)(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) AnnotatedTV_NcclAllReduce_Tz_atypes.Float32z_atypes.Float64z _atypes.Halfz _atypes.Int32z _atypes.Int64input reduction num_devices shared_namereturnc tjxstj}|j}|jr! t j |d||d|d|d| }|Stj|d}tj|d}tj|d}t!j"d|||||\} } } } | dd}tj$rjd| j'dd| j)dd| j+dd| j'df} | j,} tj.d| | ||\}|S#t j$r }tj||Yd}~nd}~wt j$rYnwxYw t||||||S#t j$rY]wxYw) aOutputs a tensor containing the reduction across all input tensors. Outputs a tensor containing the reduction across all input tensors passed to ops within the same `shared_name. The graph should be constructed so if one op runs with shared_name value `c`, then `num_devices` ops will run with shared_name value `c`. Failure to do so will cause the graph execution to fail to complete. input: the input to the reduction data: the value of the reduction across all `num_devices` devices. reduction: the reduction operation to perform. num_devices: The number of devices participating in this reduction. shared_name: Identifier that shared between ops of the same reduction. Args: input: A `Tensor`. Must be one of the following types: `half`, `float32`, `float64`, `int32`, `int64`. reduction: A `string` from: `"min", "max", "prod", "sum"`. num_devices: An `int`. shared_name: A `string`. name: A name for the operation (optional). Returns: A `Tensor`. Has the same type as `input`. NcclAllReducerrrN)rrrnamectx)rrrrrT)_contextr_thread_local_datais_eagerrTFE_Py_FastPathExecute_core_NotOkStatusException_opsraise_from_not_ok_status_FallbackExceptionnccl_all_reduce_eager_fallback_SymbolicException_executemake_strmake_int_op_def_library_apply_op_helpermust_record_gradientget_attr_get_attr_type _get_attr_intinputsrecord_gradient)rrrrr_ctxtld_resulte__op_outputs_attrs _inputs_flats R/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/ops/gen_nccl_ops.pynccl_all_reducer>s4    0h..0$ #\\ 11 otUK{M;@gn ;7)!!+}=+!!+}=+'88u %0k"$!QX QK' ""$3<< 4c  %} . ll=)+F::L vw8 (' .9  & &- ##At,,  # #    + 9+!$88  # #   s0E FE99FFF''F>=F>zraw_ops.NcclAllReducec  tj|d}tj|d}tj|d}tj|g|tj tj tjtjtjg\}\}|g}d|d|d|d|f}tjdd||||} tjrtjd||| | \} | S) Nrrrrs NcclAllReducer2attrsrrr) r)r*r+args_to_matching_eager_dtypeshalffloat32float64int32int64rr.r3) rrrrrr_attr_Tr<r;r6s r=r'r'^s ;7)!!+}=+!!+}=+55ugsW\\SZSbSbdkdsdsu|vCvCELEREREUV'8E, C- &   -q#)s ?' ""$ vw8 (' .TV_NcclBroadcast_Tctjxstj}|j}|jr t j |d||d|}|Stj|d}tj d|||\}}}} | dd}tj"rHd|j%dd|j'df} |j(} tj*d| | ||\}|S#t j$r }tj||Yd}~nd}~wt j$rYnwxYw t||||S#t j$rY wxYw)aSends `input` to all devices that are connected to the output. Sends `input` to all devices that are connected to the output. The graph should be constructed so that all ops connected to the output have a valid device assignment, and the op itself is assigned one of these devices. input: The input to the broadcast. output: The same as input. shape: The shape of the input tensor. Args: input: A `Tensor`. Must be one of the following types: `half`, `float32`, `float64`, `int32`, `int64`. shape: A `tf.TensorShape` or list of `ints`. name: A name for the operation (optional). Returns: A `Tensor`. Has the same type as `input`. NcclBroadcastshapeN)rOrr)rrOrr)rrrr rr!r"r#r$r%r&nccl_broadcast_eager_fallbackr(r) make_shaper,r-r.r0r/r2r3) rrOrr4r5r6r7r8r9r:r;r<s r=nccl_broadcastrRqs^(    0h..0$ #\\ 11 otUGU!QX QK' ""$3%%c*GS\\'5J KF::L vw8 (' .)  & &- ##At,,  # #    * u4T33  # #   s0C77D> D%%D>=D>EE('E(zraw_ops.NcclBroadcastc tj|d}tj|g|tjtj tj tjtjg\}\}|g}d|d|f}tjdd||||}tjrtjd||||\}|S)NrOrs NcclBroadcastr@rArN) r)rQrCrDrErFrGrHrIrr.r3)rrOrrrJr<r;r6s r=rPrPs   eW -%55ugsW\\SZSbSbdkdsdsu|vCvCELEREREUV'8E, '5 )&   -q#)s ?' ""$ vw8 (' .rKTV_NcclReduce_TcVtjxstj}|j}|jr t j |d||d|}|St|ttfst!d|zt#|}t%j&|d}t)j*d|||\}}} } | dd}t%j,rYd| j/dd| j1dd| j3df} | j4} t%j6d| | ||\}|S#t j$r }tj||Yd}~nd}~wt j$rYnwxYw t||||S#t j$rYKwxYw) aReduces `input` from `num_devices` using `reduction` to a single device. Reduces `input` from `num_devices` using `reduction` to a single device. The graph should be constructed so that all inputs have a valid device assignment, and the op itself is assigned one of these devices. input: The input to the reduction. data: the value of the reduction across all `num_devices` devices. reduction: the reduction operation to perform. Args: input: A list of at least 1 `Tensor` objects with the same type in: `half`, `float32`, `float64`, `int32`, `int64`. reduction: A `string` from: `"min", "max", "prod", "sum"`. name: A name for the operation (optional). Returns: A `Tensor`. Has the same type as `input`. NcclReducerN)rrr?Expected list for 'input' argument to 'nccl_reduce' Op, not %r.)rrrrr)rrrr rr!r"r#r$r%r&nccl_reduce_eager_fallbackr( isinstancelisttuple TypeErrorlenr)r*r,r-r.r/r0r1r2r3) rrrr4r5r6r7_attr_num_devicesr8r9r:r;r<s r= nccl_reducer_s(    0h..0$ #\\ 11 lD%iAg n ED%= )  $&+ , --%j ;7)'88EYTC!QX QK' ""$3<< 4c  %} .0F::L lFG5 (' .7  & &- ##At,,  # #    ' 94T;;  # #   s0D77E> E%%E>=E>FF('F(zraw_ops.NcclReducec (t|ttfstd|zt |}t j |d}t jt||tjtjtjtjtjg\}}t|}d|d|d|f}t jdd||||}t jrt j d||||\}|S) NrWrrrs NcclReducer@rArV)rYrZr[r\r]r)r*rCrDrErFrGrHrIrr.r3) rrrrr^rJr<r;r6s r=rXrXs ED%= )  $&+ , --%j ;7)224;gllT[TcTceletetv}wDwDFMFSFSFVW.'5e, C- &   ]Al#)s ?' ""$ lFG5 (' .rK)N)3__doc__ collectionstensorflow.pythonrtensorflow.python.eagerrrrr"rr)tensorflow.python.frameworkrrDtensorflow.security.fuzzing.pyr_atypesr _op_def_registryr r$r r,"tensorflow.python.util.deprecationr tensorflow.python.utilr _dispatch tensorflow.python.util.tf_exportrtypingrrrtyping_extensionsrrstrintr> to_raw_oprr'rLrRrNrPrTr_rVrXrKr=rss 67179FK3IC86%%'13DFWYgix{JK>9S*<%<=>#>\_>nq>BKLOQcLcBd>@3 12>4>>/3RS )C9K4K*LY\kn~AQZ[^`r[rQs"13DFWYgix{JK/)C);$;</S\]`bt]tSu/b3 12>4>>.3QR  38J3J)K bklorDmDcE +->@QSacruDE6ycO!;<66\efikzfz\{6p-Y+ ,^T^^K-H I iS ?0J&KX[ktux{JvJlKrK