o ®Ÿ?e%ã@sü dZddlZddlmZddlmZddlmZddlm Z ddl m Z ddlmZdd l mZdd l mZdd l mZdd lmZdd lmZddlmZddlmZm Z edej!ej"ej#ej$ej%ej&ƒZ'dydej(e'dej(ej%dej(ej%dej(ej%de ej(ej)de*de+dej(e'fdd„Z,edƒe -e,¡ƒZ.dej(e'dej(ej%dej(ej%dej(ej%de ej(ej)de*de+dej(e'fdd„Z/edej!ej"ej#ej$ej%ej&ƒZ0dzdej(e0d ej(ej)d!ej(ej%de+dej(e0f d"d#„Z1ed$ƒe -e1¡ƒZ2dej(e0d ej(ej)d!ej(ej%de+dej(e0f d%d&„Z3e 4d'ddg¡Z5d{d!ej(ej%d(ej(ej%d)ej(ej%fd*d+„Z6ed,ƒe -e6¡ƒZ7d!ej(ej%d(ej(ej%d)ej(ej%fd-d.„Z8ed/ej9ej"ej#ej$ej%ej&ƒZ:dyd0e:de;de;de;de*de+dej(e:fd1d2„Zed6ej9ej"ej#ej$ej%ej&ƒZ?ed7ej%ej&ƒZ@dydej(ej%dej(ej%dej(ej%d8ej(e@d0e?de*de+dej(e?fd9d:„ZAed;ƒe -eA¡ƒZBdej(ej%dej(ej%dej(ej%d8ej(e@d0e?de*de+dej(e?fdej9ej"ej#ej$ej%ej&ƒZDdydej(eDde;de;de;de*de+dej(eDfd?d@„ZEedAƒe -eE¡ƒZFdej(eDde;de;de;de*de+dej(eDfdBdC„ZGedDej9ej"ej#ej$ej%ej&ƒZHdydej(eHdej(ej%dej(ej%dej(ej%de*de+dej(eHfdEdF„ZIedGƒe -eI¡ƒZJdej(eHdej(ej%dej(ej%dej(ej%de*de+dej(eHfdHdI„ZKedJej"ej#ej$ej%ej&ƒZLdydej(eLde;de;de;de*de+dej(eLfdKdL„ZMedMƒe -eM¡ƒZNdej(eLde;de;de;de*de+dej(eLfdNdO„ZOedPej"ej#ej$ej%ej&ƒZPdydej(ePdej(ej%dej(ej%dej(ej%de ej(ej)de*de+dej(ePfdQdR„ZQedSƒe -eQ¡ƒZRdej(ePdej(ej%dej(ej%dej(ej%de ej(ej)de*de+dej(ePfdTdU„ZSdydej(ej%dVej(ej%dej(ej%de*de+dej(ej)f dWdX„ZTedYƒe -eT¡ƒZUdej(ej%dVej(ej%dej(ej%de*de+dej(ej)f dZd[„ZVed\ej!ej"ej#ej$ej%ej&ƒZWgdddfdej(eWde;de;de;d]e*d^e*de*de+dej(eWfd_d`„ZXedaƒe -eX¡ƒZYdej(eWde;de;de;d]e*d^e*de*de+dej(eWfdbdc„ZZeddej!ej"ej#ej$ej%ej&ƒZ[d|dej(e[dej(ej%dej(ej%dej(ej%de ej(ej)d]e*d^e*de*de+dfe;dej(e[fdgdh„Z\ediƒe -e\¡ƒZ]dej(e[dej(ej%dej(ej%dej(ej%de ej(ej)d]e*d^e*de*de+dfe;dej(e[fdjdk„Z^edlej!ej"ej#ej$ej%ej&ƒZ_d|dej(e_dej(ej%dej(ej%dej(ej%de ej(ej)d]e*d^e*de*de+dfe;dej(e_fdmdn„Z`edoƒe -e`¡ƒZadej(e_dej(ej%dej(ej%dej(ej%de ej(ej)d]e*d^e*de*de+dfe;dej(e_fdpdq„Zbedrej!ej"ej#ej$ej%ej&ƒZcdzdej(ecd ej(ej)d!ej(ej%dse*de+dej(ecf dtdu„Zdedvƒe -ed¡ƒZedej(ecd ej(ej)d!ej(ej%dse*de+dej(ecf dwdx„ZfdS)}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)ÚTypeVarÚListÚTV_CollectiveAllToAllV2_TÚautoÚinputÚ group_sizeÚ group_keyÚ instance_keyÚordering_tokenÚcommunication_hintÚtimeout_secondsÚreturncCóŽtjpt ¡}|j} | jr\zt |d||||||d|d|¡ } | WStjy9} z t  | |¡WYd} ~ nd} ~ wtj yBYnwzt |||||||||d WStj y[Ynwt |ttfƒsitd|ƒ‚t|ƒ} |dursd}t |d¡}|durd}t |d¡}tjd||||||||d \} } }}|dd…} t ¡rÂd | d ¡d| d¡d| d¡d | d ¡f}|j}t d||| ¡| \} | S) auMutually exchanges multiple tensors of identical type and shape. Args: input: A `Tensor`. Must be one of the following types: `bfloat16`, `float32`, `half`, `float64`, `int32`, `int64`. group_size: A `Tensor` of type `int32`. group_key: A `Tensor` of type `int32`. instance_key: A `Tensor` of type `int32`. ordering_token: A list of `Tensor` objects with type `resource`. communication_hint: An optional `string`. Defaults to `"auto"`. timeout_seconds: An optional `float`. Defaults to `0`. name: A name for the operation (optional). Returns: A `Tensor`. Has the same type as `input`. ÚCollectiveAllToAllV2rrN©rrÚnameÚctxúUExpected list for 'ordering_token' argument to 'collective_all_to_all_v2' Op, not %r.rr©rrrrrrrrÚTÚNordering_token)Ú_contextrÚ_thread_local_dataÚis_eagerrÚTFE_Py_FastPathExecuteÚ_coreÚ_NotOkStatusExceptionÚ_opsÚraise_from_not_ok_statusÚ_FallbackExceptionÚ'collective_all_to_all_v2_eager_fallbackÚ_SymbolicExceptionÚ isinstanceÚlistÚtupleÚ TypeErrorÚlenÚ_executeÚmake_strÚ make_floatÚ_op_def_libraryÚ_apply_op_helperÚmust_record_gradientÚ_get_attr_typeÚget_attrÚ _get_attr_intÚinputsÚrecord_gradient©rrrrrrrrÚ_ctxÚtldÚ_resultÚeÚ_attr_Nordering_tokenÚ_Ú_opÚ_outputsÚ_attrsÚ _inputs_flat©rIúi/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/ops/gen_collective_ops.pyÚcollective_all_to_all_v2sv ý€ÿ ýÿÿÿ  ú    ýÿrKzraw_ops.CollectiveAllToAllV2c  Cst|ttfƒs td|ƒ‚t|ƒ} |durd}t |d¡}|dur#d}t |d¡}t |g|t j t j t j t j t jt jg¡\} \}t |t j¡}t |t j¡}t |t j¡}t |t j¡}||||gt|ƒ} d| d|d|d| f} tjdd | | ||d } t ¡r‡t d | | | ¡| \} | S) Nrrrrrr!r"sCollectiveAllToAllV2é©r<Úattrsrrr)r.r/r0r1r2r3r4r5Úargs_to_matching_eagerÚ_dtypesÚbfloat16Úfloat32ÚhalfÚfloat64Úint32Úint64r)Úconvert_to_tensorÚconvert_n_to_tensorÚresourcerr8r=©rrrrrrrrrrCÚ_attr_TrHrGrArIrIrJr,`s@ÿÿ  .þ ÿÿr,ÚTV_CollectiveAllToAllV3_TÚ communicatorÚgroup_assignmentc Cs,tjpt ¡}|j}|jrUzt |d||||d|¡}|WStjy5}z t  ||¡WYd}~nd}~wtj y>Ynwz t ||||||dWStj yTYnw|dur[d}t  |d¡}tjd|||||d\} } } } | dd…}t  ¡r‘d|  d¡d|  d¡f} | j} t  d| | |¡|\}|S)aÊMutually exchanges multiple tensors of identical type and shape. Args: input: A `Tensor`. Must be one of the following types: `bfloat16`, `float32`, `half`, `float64`, `int32`, `int64`. communicator: A `Tensor` of type `resource`. group_assignment: A `Tensor` of type `int32`. timeout_seconds: An optional `float`. Defaults to `0`. name: A name for the operation (optional). Returns: A `Tensor`. Has the same type as `input`. ÚCollectiveAllToAllV3rN)rrrr)rr]r^rrr!)r#rr$r%rr&r'r(r)r*r+Ú'collective_all_to_all_v3_eager_fallbackr-r3r5r6r7r8r9r:r<r=)rr]r^rrr?r@rArBrDrErFrGrHrIrIrJÚcollective_all_to_all_v3€sT  þ€ÿþÿ ý  ÿÿrazraw_ops.CollectiveAllToAllV3c Cs°|durd}t |d¡}t |g|tjtjtjtjtjtj g¡\}\}t   |tj ¡}t   |tj¡}|||g}d|d|f}tj dd||||d} t ¡rSt d||| ¡| \} | S)Nrrr!sCollectiveAllToAllV3rLrMr_)r3r5rOrPrQrRrSrTrUrVr)rWrYrr8r=) rr]r^rrrr[rHrGrArIrIrJr`´s" .   ÿÿr`ÚCollectiveAssignGroupV2Ú device_indexÚbase_keyc Cstjpt ¡}|j}|jrWzt |d||||¡}t |¡}|WStj y8}z t   ||¡WYd}~nd}~wtj yAYnwz t |||||dWStjyVYnwtjd||||d\}}} } | dd…}t ¡r|d} | j} t d| | |¡t |¡}|S)aŽAssign group keys based on group assignment. Args: group_assignment: A `Tensor` of type `int32`. device_index: A `Tensor` of type `int32`. base_key: A `Tensor` of type `int32`. name: A name for the operation (optional). Returns: A tuple of `Tensor` objects (group_size, group_key). group_size: A `Tensor` of type `int32`. group_key: A `Tensor` of type `int32`. rbN)rr)r^rcrdrrI)r#rr$r%rr&Ú_CollectiveAssignGroupV2OutputÚ_maker'r(r)r*r+Ú)collective_assign_group_v2_eager_fallbackr-r6r7r3r8r<r=) r^rcrdrr?r@rArBrDrErFrGrHrIrIrJÚcollective_assign_group_v2ÊsJ þ €ÿ ÿÿý ÿ rhzraw_ops.CollectiveAssignGroupV2cCstt |tj¡}t |tj¡}t |tj¡}|||g}d}tjdd||||d}t ¡r3t d|||¡t  |¡}|S)NsCollectiveAssignGroupV2érMrb) r)rWrPrUr3rr8r=rerf)r^rcrdrrrHrGrArIrIrJrgüs þÿ rgÚTV_CollectiveBcastRecv_Tr!cCsÐtjpt ¡}|j} | jrazt |d|d|d|d|d|d|d|d|¡} | WStjy>} z t  | |¡WYd } ~ nd } ~ wtj yGYnwzt |||||||||d WStj y`Ynwt  |d¡}t  |d¡}t  |d¡}t  |d¡}t  |d¡}|d ur…d }t  |d¡}|d ur‘d }t  |d¡}tjd||||||||d \} } } }|d d …} t  ¡rãd|  d¡d|  d¡d|  d¡d|  d¡d|  d¡d|  d¡d|  d¡f}| j}t  d||| ¡| \} | S)aòReceives a tensor value broadcast from another device. Args: T: A `tf.DType` from: `tf.bool, tf.float32, tf.half, tf.float64, tf.int32, tf.int64`. group_size: An `int`. group_key: An `int`. instance_key: An `int`. shape: A `tf.TensorShape` or list of `ints`. communication_hint: An optional `string`. Defaults to `"auto"`. timeout_seconds: An optional `float`. Defaults to `0`. name: A name for the operation (optional). Returns: A `Tensor` of type `T`. ÚCollectiveBcastRecvr!rrrÚshaperrN) r!rrrrlrrrrrr)r!rrrrlrrr)r#rr$r%rr&r'r(r)r*r+Ú$collective_bcast_recv_eager_fallbackr-r3Ú make_typeÚmake_intÚ make_shaper4r5r6r7r8r9r;r:r<r=)r!rrrrlrrrr?r@rArBrDrErFrGrHrIrIrJÚcollective_bcast_recvsz ü€ÿüÿ       û       úÿrqzraw_ops.CollectiveBcastRecvc CsÈt |d¡}t |d¡}t |d¡}t |d¡}t |d¡}|dur$d}t |d¡}|dur0d}t |d ¡}g} d|d|d|d|d|d|d |f} tjd d | | ||d } t ¡r_t d | | | ¡| \} | S)Nr!rrrrlrrrrsCollectiveBcastRecvrLrMrk) r3rnrorpr4r5rr8r=) r!rrrrlrrrrrHrGrArIrIrJrmXs2         þ ÿÿrmÚTV_CollectiveBcastRecvV2_TÚTV_CollectiveBcastRecvV2_TshaperlcCsztjpt ¡}|j} | jr]zt |d|||||d|d|d|¡ } | WStjy:} z t  | |¡WYd} ~ nd} ~ wtj yCYnwzt |||||||||d WStj y\Ynwt  |d¡}|durid}t  |d¡}|durud}t  |d¡}tjd||||||||d \} } } }|dd…} t  ¡r¸d|  d¡d |  d ¡d|  d¡d|  d¡f}| j}t  d||| ¡| \} | S) aDReceives a tensor value broadcast from another device. Args: group_size: A `Tensor` of type `int32`. group_key: A `Tensor` of type `int32`. instance_key: A `Tensor` of type `int32`. shape: A `Tensor`. Must be one of the following types: `int32`, `int64`. T: A `tf.DType` from: `tf.bool, tf.float32, tf.half, tf.float64, tf.int32, tf.int64`. communication_hint: An optional `string`. Defaults to `"auto"`. timeout_seconds: An optional `float`. Defaults to `0`. name: A name for the operation (optional). Returns: A `Tensor` of type `T`. ÚCollectiveBcastRecvV2r!rrN)r!rrrrrr)rrrrlr!rrrÚTshape)r#rr$r%rr&r'r(r)r*r+Ú'collective_bcast_recv_v2_eager_fallbackr-r3rnr4r5r6r7r8r9r:r<r=)rrrrlr!rrrr?r@rArBrDrErFrGrHrIrIrJÚcollective_bcast_recv_v2tsf  ý€ÿ ýÿ   ü    ýÿrwzraw_ops.CollectiveBcastRecvV2c Csàt |d¡}|dur d}t |d¡}|durd}t |d¡}t |g|tjtjgtj¡\} \}t  |tj¡}t  |tj¡}t  |tj¡}||||g} d|d| d|d|f} tj dd| | ||d } t  ¡rkt  d | | | ¡| \} | S) Nr!rrrrrusCollectiveBcastRecvV2rLrMrt) r3rnr4r5rOrPrUrVr)rWrr8r=) rrrrlr!rrrrZ _attr_TshaperHrGrArIrIrJrv´s0   "  ÿ ÿÿrvÚTV_CollectiveBcastSend_TcCóÂtjpt ¡}|j} | jr`zt |d||d|d|d|d|d|d|¡} | WStjy=} z t  | |¡WYd} ~ nd} ~ wtj yFYnwzt |||||||||d WStj y_Ynwt  |d¡}t  |d¡}t  |d¡}t  |d¡}|dur~d }t  |d¡}|durŠd }t  |d¡}tjd||||||||d \} } } }|dd…} t  ¡rÜd |  d ¡d|  d¡d|  d¡d|  d¡d|  d¡d|  d¡d|  d¡f}| j}t  d||| ¡| \} | S)aBroadcasts a tensor value to one or more other devices. Args: input: A `Tensor`. Must be one of the following types: `bool`, `float32`, `half`, `float64`, `int32`, `int64`. group_size: An `int`. group_key: An `int`. instance_key: An `int`. shape: A `tf.TensorShape` or list of `ints`. communication_hint: An optional `string`. Defaults to `"auto"`. timeout_seconds: An optional `float`. Defaults to `0`. name: A name for the operation (optional). Returns: A `Tensor`. Has the same type as `input`. ÚCollectiveBcastSendrrrrlrrN©rrrrlrrrrrr©rrrrrlrrrr!)r#rr$r%rr&r'r(r)r*r+Ú$collective_bcast_send_eager_fallbackr-r3rorpr4r5r6r7r8r9r;r:r<r=©rrrrrlrrrr?r@rArBrDrErFrGrHrIrIrJÚcollective_bcast_sendÎóx  ü€ÿüÿ      û       úÿrzraw_ops.CollectiveBcastSendc Csìt |d¡}t |d¡}t |d¡}t |d¡}|durd}t |d¡}|dur*d}t |d¡}t |g|tjtjtj tj tj tj g¡\} \}|g} d | d|d|d|d|d|d|f} tj d d | | ||d } t ¡rqt d | | | ¡| \} | S)Nrrrrlrrrrr!sCollectiveBcastSendrLrMrz)r3rorpr4r5rOrPÚboolrRrSrTrUrVrr8r=© rrrrrlrrrrr[rHrGrArIrIrJr}s2      .  þ ÿÿr}ÚTV_CollectiveBcastSendV2_Tc Cs\tjpt ¡}|j}|jrZzt |d|||||d|d|¡ } | WStjy8} z t  | |¡WYd} ~ nd} ~ wtj yAYnwz t ||||||||dWStj yYYnw|dur`d}t  |d¡}|durld}t  |d¡}tjd|||||||d\} } } } | dd…} t  ¡r©d |  d ¡d|  d¡d|  d¡f}| j}t  d||| ¡| \} | S) a#Broadcasts a tensor value to one or more other devices. Args: input: A `Tensor`. Must be one of the following types: `bool`, `float32`, `half`, `float64`, `int32`, `int64`. group_size: A `Tensor` of type `int32`. group_key: A `Tensor` of type `int32`. instance_key: A `Tensor` of type `int32`. communication_hint: An optional `string`. Defaults to `"auto"`. timeout_seconds: An optional `float`. Defaults to `0`. name: A name for the operation (optional). Returns: A `Tensor`. Has the same type as `input`. ÚCollectiveBcastSendV2rrNrrr)rrrrrrrr!)r#rr$r%rr&r'r(r)r*r+Ú'collective_bcast_send_v2_eager_fallbackr-r3r4r5r6r7r8r9r:r<r=)rrrrrrrr?r@rArBrDrErFrGrHrIrIrJÚcollective_bcast_send_v22sd ý€ÿýÿ  û   þÿr†zraw_ops.CollectiveBcastSendV2c CsÜ|durd}t |d¡}|durd}t |d¡}t |g|tjtjtjtjtj tj g¡\}\}t   |tj ¡}t   |tj ¡}t   |tj ¡}||||g} d|d|d|f} tj dd| | ||d} t ¡rit d | | | ¡| \} | S) Nrrrrr!sCollectiveBcastSendV2rLrMr„)r3r4r5rOrPrrRrSrTrUrVr)rWrr8r=) rrrrrrrrr[rHrGrArIrIrJr…ps.  . ÿ ÿÿr…ÚTV_CollectiveGather_TcCry)a!Mutually accumulates multiple tensors of identical type and shape. Args: input: A `Tensor`. Must be one of the following types: `float32`, `half`, `float64`, `int32`, `int64`. group_size: An `int`. group_key: An `int`. instance_key: An `int`. shape: A `tf.TensorShape` or list of `ints`. communication_hint: An optional `string`. Defaults to `"auto"`. timeout_seconds: An optional `float`. Defaults to `0`. name: A name for the operation (optional). Returns: A `Tensor`. Has the same type as `input`. ÚCollectiveGatherrrrrlrrNr{rrr|r!)r#rr$r%rr&r'r(r)r*r+Ú collective_gather_eager_fallbackr-r3rorpr4r5r6r7r8r9r;r:r<r=r~rIrIrJÚcollective_gather‰r€rŠzraw_ops.CollectiveGatherc Csèt |d¡}t |d¡}t |d¡}t |d¡}|durd}t |d¡}|dur*d}t |d¡}t |g|tjtjtj tj tj g¡\} \}|g} d | d|d|d|d|d|d|f} tj d d | | ||d } t  ¡rot d | | | ¡| \} | S)Nrrrrlrrrrr!sCollectiveGatherrLrMrˆ)r3rorpr4r5rOrPrRrSrTrUrVrr8r=r‚rIrIrJr‰Òs2      *  þ ÿÿr‰ÚTV_CollectiveGatherV2_TcCr) akMutually accumulates multiple tensors of identical type and shape. Args: input: A `Tensor`. Must be one of the following types: `float32`, `half`, `float64`, `int32`, `int64`. group_size: A `Tensor` of type `int32`. group_key: A `Tensor` of type `int32`. instance_key: A `Tensor` of type `int32`. ordering_token: A list of `Tensor` objects with type `resource`. communication_hint: An optional `string`. Defaults to `"auto"`. timeout_seconds: An optional `float`. Defaults to `0`. name: A name for the operation (optional). Returns: A `Tensor`. Has the same type as `input`. ÚCollectiveGatherV2rrNrúQExpected list for 'ordering_token' argument to 'collective_gather_v2' Op, not %r.rrr r!r")r#rr$r%rr&r'r(r)r*r+Ú#collective_gather_v2_eager_fallbackr-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=r>rIrIrJÚcollective_gather_v2íst ý€ÿ ýÿÿÿ  û    ýÿrzraw_ops.CollectiveGatherV2c  Cst|ttfƒs td|ƒ‚t|ƒ} |durd}t |d¡}|dur#d}t |d¡}t |g|t j t j t j t j t jg¡\} \}t |t j ¡}t |t j ¡}t |t j ¡}t |t j¡}||||gt|ƒ} d| d|d|d| f} tjdd | | ||d } t ¡r…t d | | | ¡| \} | S) Nrrrrrr!r"sCollectiveGatherV2rLrMrŒ)r.r/r0r1r2r3r4r5rOrPrRrSrTrUrVr)rWrXrYrr8r=rZrIrIrJrŽ2s@ÿÿ  *þ ÿÿrŽÚrankc CsLtjpt ¡}|j}|jrXzt |d||||d|d|¡ }|WStjy7} z t  | |¡WYd} ~ nd} ~ wtj y@Ynwz t |||||||dWStj yWYnw|dur^d}t  |d¡}|durjd}t  |d¡}tjd||||||d\} } } } | dd…}t  ¡r¡d|  d¡d|  d¡f} | j}t  d|| |¡|\}|S) a”Initializes a group for collective operations. Args: group_key: A `Tensor` of type `int32`. rank: A `Tensor` of type `int32`. group_size: A `Tensor` of type `int32`. communication_hint: An optional `string`. Defaults to `"auto"`. timeout_seconds: An optional `float`. Defaults to `0`. name: A name for the operation (optional). Returns: A `Tensor` of type `resource`. Ú CollectiveInitializeCommunicatorrrNrrr)rrrrrr)r#rr$r%rr&r'r(r)r*r+Ú1collective_initialize_communicator_eager_fallbackr-r3r4r5r6r7r8r:r<r=)rrrrrrr?r@rArBrDrErFrGrHrIrIrJÚ"collective_initialize_communicatorPs` ý€ÿþÿ  û   ÿÿr“z(raw_ops.CollectiveInitializeCommunicatorc Cs¨|durd}t |d¡}|durd}t |d¡}t |tj¡}t |tj¡}t |tj¡}|||g}d|d|f}tjdd||||d} t ¡rOt  d||| ¡| \} | S) Nrrrrs CollectiveInitializeCommunicatorrLrMr‘) r3r4r5r)rWrPrUrr8r=) rrrrrrrrHrGrArIrIrJr’‹s.   ÿþÿr’ÚTV_CollectiveReduce_TÚmerge_opÚfinal_opc Csbtjpt ¡} | j} | jrizt | d| |d|d|d|d|d|d|d|d |d | ¡} | WStjyC}z t  || ¡WYd }~nd }~wtj yLYnwzt |||||||||| | | d WStj yhYnwt  |d¡}t  |d¡}t  |d¡}t  |d¡}t  |d¡}t|ttfƒs”td |ƒ‚dd„|Dƒ}|d ur¡g}t|ttfƒs®td|ƒ‚dd„|Dƒ}|d ur»d}t  |d ¡}| d urÇd} t  | d ¡} tjd|||||||||| | d \}}}}|d d …} t  ¡r,d| d¡d| d¡d| d¡d| d¡d| d¡d| d¡d| d¡d| d¡d | d ¡d | d ¡f}|j}t  d||| ¡| \} | S)aÅMutually reduces multiple tensors of identical type and shape. Args: input: A `Tensor`. Must be one of the following types: `bfloat16`, `float32`, `half`, `float64`, `int32`, `int64`. group_size: An `int`. group_key: An `int`. instance_key: An `int`. merge_op: A `string` from: `"Min", "Max", "Mul", "Add"`. final_op: A `string` from: `"Id", "Div"`. subdiv_offsets: A list of `ints`. wait_for: An optional list of `ints`. Defaults to `[]`. communication_hint: An optional `string`. Defaults to `"auto"`. timeout_seconds: An optional `float`. Defaults to `0`. name: A name for the operation (optional). Returns: A `Tensor`. Has the same type as `input`. ÚCollectiveReducerrrr•r–Úsubdiv_offsetsÚwait_forrrN) rrrr•r–r˜r™rrrrúNExpected list for 'subdiv_offsets' argument to 'collective_reduce' Op, not %r.cSóg|]}t |d¡‘qS©r˜©r3ro©Ú.0Z_irIrIrJÚ Ùóz%collective_reduce..úHExpected list for 'wait_for' argument to 'collective_reduce' Op, not %r.cSr›©r™rržrIrIrJr àr¡rr) rrrrr•r–r˜r™rrrr!)r#rr$r%rr&r'r(r)r*r+Ú collective_reduce_eager_fallbackr-r3ror4r.r/r0r1r5r6r7r8r9r;r:r<r=)rrrrr•r–r˜r™rrrr?r@rArBrDrErFrGrHrIrIrJÚcollective_reduce¤s¤   û€ÿûÿ     ÿÿÿÿ  ú         ø ÿr¥zraw_ops.CollectiveReducec Cs`t |d¡}t |d¡}t |d¡}t |d¡}t |d¡}t|ttfƒs+td|ƒ‚dd„|Dƒ}|dur8g}t|ttfƒsEtd |ƒ‚d d„|Dƒ}|durRd }t |d ¡}| dur^d } t | d¡} t |g| t j t j t j t j t jt jg¡\} \}|g} d| d|d|d|d|d|d|d|d |d| f}tjdd| || | d}t ¡r«t d| ||¡|\}|S)Nrrrr•r–ršcSr›rœrržrIrIrJr  r¡z4collective_reduce_eager_fallback..r¢cSr›r£rržrIrIrJr r¡rrrrr!r˜r™sCollectiveReducerLrMr—)r3ror4r.r/r0r1r5rOrPrQrRrSrTrUrVrr8r=)rrrrr•r–r˜r™rrrrr[rHrGrArIrIrJr¤sX     ÿÿÿÿ  .  ü ÿÿr¤ÚTV_CollectiveReduceScatterV2_TéÿÿÿÿÚmax_subdivs_per_devicec Cóôtjpt ¡} | j} | jrezt | d| |||||d|d|d|d|d| ¡} | WStjy?}z t  || ¡WYd}~nd}~wtj yHYnwzt |||||||||| | | d WStj ydYnwt |ttfƒsrtd |ƒ‚t|ƒ}t |d¡}t |d¡}|durˆd }t |d¡}|dur”d }t |d¡}| dur d } t | d¡} tjd|||||||||| | d \}}}}|dd…} t ¡rõd| d¡d| d¡d| d¡d| d¡d| d¡d| d¡d| d¡f}|j}t d||| ¡| \} | S)a7Mutually reduces multiple tensors of identical type and shape and scatters the result. Args: input: A `Tensor`. Must be one of the following types: `bfloat16`, `float32`, `half`, `float64`, `int32`, `int64`. group_size: A `Tensor` of type `int32`. group_key: A `Tensor` of type `int32`. instance_key: A `Tensor` of type `int32`. ordering_token: A list of `Tensor` objects with type `resource`. merge_op: A `string` from: `"Min", "Max", "Mul", "Add"`. final_op: A `string` from: `"Id", "Div"`. communication_hint: An optional `string`. Defaults to `"auto"`. timeout_seconds: An optional `float`. Defaults to `0`. max_subdivs_per_device: An optional `int`. Defaults to `-1`. name: A name for the operation (optional). Returns: A `Tensor`. Has the same type as `input`. ÚCollectiveReduceScatterV2r•r–rrr¨N©r•r–rrr¨rrúYExpected list for 'ordering_token' argument to 'collective_reduce_scatter_v2' Op, not %r.rrr§© rrrrrr•r–rrr¨rr!r")r#rr$r%rr&r'r(r)r*r+Ú+collective_reduce_scatter_v2_eager_fallbackr-r.r/r0r1r2r3r4r5ror6r7r8r9r:r;r<r=©rrrrrr•r–rrr¨rr?r@rArBrCrDrErFrGrHrIrIrJÚcollective_reduce_scatter_v2,s’  ü€ÿ ûÿÿÿ     ÷     úÿr°z!raw_ops.CollectiveReduceScatterV2c CóTt|ttfƒs td|ƒ‚t|ƒ} t |d¡}t |d¡}|dur#d}t |d¡}|dur/d}t |d¡}| dur;d} t | d ¡} t  |g| t j t j t j t jt jt jg¡\} \}t |t j¡}t |t j¡}t |t j¡}t |t j¡}||||gt|ƒ}d | d|d|d|d|d | d | f}tjd d ||| | d}t ¡r¥t d|||¡|\}|S)Nr¬r•r–rrrrr§r¨r!r"sCollectiveReduceScatterV2rLrMrª©r.r/r0r1r2r3r4r5rorOrPrQrRrSrTrUrVr)rWrXrYrr8r=©rrrrrr•r–rrr¨rrrCr[rHrGrArIrIrJr®ƒsNÿÿ     . ýþÿr®ÚTV_CollectiveReduceV2_Tc Cr©)aMutually reduces multiple tensors of identical type and shape. Args: input: A `Tensor`. Must be one of the following types: `bfloat16`, `float32`, `half`, `float64`, `int32`, `int64`. group_size: A `Tensor` of type `int32`. group_key: A `Tensor` of type `int32`. instance_key: A `Tensor` of type `int32`. ordering_token: A list of `Tensor` objects with type `resource`. merge_op: A `string` from: `"Min", "Max", "Mul", "Add"`. final_op: A `string` from: `"Id", "Div"`. communication_hint: An optional `string`. Defaults to `"auto"`. timeout_seconds: An optional `float`. Defaults to `0`. max_subdivs_per_device: An optional `int`. Defaults to `-1`. name: A name for the operation (optional). Returns: A `Tensor`. Has the same type as `input`. ÚCollectiveReduceV2r•r–rrr¨Nr«úQExpected list for 'ordering_token' argument to 'collective_reduce_v2' Op, not %r.rrr§r­r!r")r#rr$r%rr&r'r(r)r*r+Ú#collective_reduce_v2_eager_fallbackr-r.r/r0r1r2r3r4r5ror6r7r8r9r:r;r<r=r¯rIrIrJÚcollective_reduce_v2ªs  ü€ÿ ûÿÿÿ     ø     úÿr¸zraw_ops.CollectiveReduceV2c Cr±)Nr¶r•r–rrrrr§r¨r!r"sCollectiveReduceV2rLrMrµr²r³rIrIrJr·sLÿÿ     . ý ÿÿr·ÚTV_CollectiveReduceV3_TÚ reductionc CsJtjpt ¡}|j}|jrXzt |d||||d|d|¡ }|WStjy7} z t  | |¡WYd} ~ nd} ~ wtj y@Ynwz t |||||||dWStj yWYnwt  |d¡}|durdd}t  |d¡}tjd||||||d\} } } } | dd…}t  ¡r d|  d¡d|  d¡d|  d¡f} | j}t  d|| |¡|\}|S) aMutually reduces multiple tensors of identical type and shape. Args: input: A `Tensor`. Must be one of the following types: `bfloat16`, `float32`, `half`, `float64`, `int32`, `int64`. communicator: A `Tensor` of type `resource`. group_assignment: A `Tensor` of type `int32`. reduction: A `string` from: `"Min", "Max", "Mul", "Add"`. timeout_seconds: An optional `float`. Defaults to `0`. name: A name for the operation (optional). Returns: A `Tensor`. Has the same type as `input`. ÚCollectiveReduceV3rºrN)rºrrrr)rr]r^rºrrr!)r#rr$r%rr&r'r(r)r*r+Ú#collective_reduce_v3_eager_fallbackr-r3r4r5r6r7r8r9r:r<r=)rr]r^rºrrr?r@rArBrDrErFrGrHrIrIrJÚcollective_reduce_v3&s\ ý€ÿþÿ  ü   þÿr½zraw_ops.CollectiveReduceV3c CsÀt |d¡}|dur d}t |d¡}t |g|tjtjtjtjtj tj g¡\}\}t   |tj ¡}t   |tj ¡}|||g}d|d|d|f} tjdd|| ||d} t ¡r[t d|| | ¡| \} | S) Nrºrrr!sCollectiveReduceV3rLrMr»)r3r4r5rOrPrQrRrSrTrUrVr)rWrYrr8r=) rr]r^rºrrrr[rHrGrArIrIrJr¼_s(  .  ÿ ÿÿr¼)rrN)rN)N)rrr§N)gÚ__doc__Ú collectionsZtensorflow.pythonrZtensorflow.python.eagerrr#rr'rr3Ztensorflow.python.frameworkrrPZtensorflow.security.fuzzing.pyrZ_atypesrZ_op_def_registryr r)r r6Z"tensorflow.python.util.deprecationr Ztensorflow.python.utilr Ú _dispatchZ tensorflow.python.util.tf_exportr ÚtypingrrZBFloat16ZFloat32ZFloat64ZHalfZInt32ZInt64rZTensorFuzzingAnnotationÚResourceÚstrÚfloatrKZ to_raw_oprr,r\rar_r`Ú namedtuplererhrbrgZBoolrjÚintrqrkrmrrrsrwrtrvrxrrzr}rƒr†r„r…r‡rŠrˆr‰r‹rrŒrŽr“r‘r’r”r¥r—r¤r¦r°rªr®r´r¸rµr·r¹r½r»r¼rIrIrIrJÚs”             \CZ <1:þ0/. .G, P=N 4F2 L;J4F2\BZD8B D[: (hTf %hSf $@6B