o ?e]@sdZddlZddlmZddlmZddlmZddlm Z ddl m Z ddlmZdd l mZdd l mZdd l mZdd lmZdd lmZddlmZddlmZm Z e!dgdZ"edej#ej$ej%ej&ej'ej(ej)Z*d;dej+e*dej+e*de,de-fddZ.ede/e.Z0dej+e*dej+e*de,de-fddZ1e!dgdZ2edej#ej$ej%ej&ej'ej(ej)Z3d;dej+e3dej+ej%d ej+e3d!ej+ej%de,de-f d"d#Z4ed$e/e4Z5dej+e3dej+ej%d ej+e3d!ej+ej%de,de-f d%d&Z6ed'ej#ej$ej%ej&ej'ej(ej)Z7d;d(ej+ej%d)ej+e7d*ej+ej%de-d+ej+ej$f d,d-Z8ed.e/e8Z9d(ej+ej%d)ej+e7d*ej+ej%de-d+ej+ej$f d/d0Z:e!d1gdZ;ed2ej#ej$ej%ej&ej'ej(ej)Zd3ej+ej%d4ej+e<d5ej+ej%dej+ej%d ej+e<d!ej+ej%de,de-fd9d:Z?dS)r?rL)r'r(r)rArBrCrDrErFrGrHrIr"convert_to_tensorrr,r1rQr) rrNrOrPrrrrrJrKr:r9r4r;r;r<rRs,  2  rR TV_SetSize_T set_indices set_values set_shapereturnc 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 rd| dd| df} | j} t d| | ||\}|S)aNumber of unique elements along last dimension of input `set`. Input `set` is a `SparseTensor` represented by `set_indices`, `set_values`, and `set_shape`. The last dimension contains values in a set, duplicates are allowed but ignored. If `validate_indices` is `True`, this op validates the order and range of `set` indices. Setting is to `False` while passing invalid arguments results in undefined behavior. Args: set_indices: A `Tensor` of type `int64`. 2D `Tensor`, indices of a `SparseTensor`. set_values: A `Tensor`. Must be one of the following types: `int8`, `int16`, `int32`, `int64`, `uint8`, `uint16`, `string`. 1D `Tensor`, values of a `SparseTensor`. set_shape: A `Tensor` of type `int64`. 1D `Tensor`, shape of a `SparseTensor`. validate_indices: An optional `bool`. Defaults to `True`. name: A name for the operation (optional). Returns: A `Tensor` of type `int32`. SetSizerN)rrrT)rVrWrXrrr)rrrrrrr r!r"r#r$set_size_eager_fallbackr&r'r)r*r+r,r.r/r0r1)rVrWrXrrr2r3r4r5r6r7r8r9r:r;r;r<set_sizesT    r\zraw_ops.SetSizec Cs|durd}t|d}t|g|tjtjtjtjtjtj tj g\}\}t |tj}t |tj}|||g}d|d|f}tj dd||||d} trUtd||| | \} | S)NTrrsSetSizer?rZ)r'r)rArBrCrDrErFrGrHrIr"rTrr,r1) rVrWrXrrrrJr:r9r4r;r;r<r[2s" 2   r[SparseToSparseSetOperationTV_SparseToSparseSetOperation_T set1_indices set1_values set1_shapec Csjtjpt} | j} | jrczt| d|||||||d|d| } t| } | WStj y?} z t | |WYd} ~ nd} ~ wtj yHYnwzt |||||||||| d WStjybYnwt|d}|durod}t|d}tjd|||||||||d \} } }}|dd} trd|dd|dd|df}|j}td||| t| } | S) a Applies set operation along last dimension of 2 `SparseTensor` inputs. See SetOperationOp::SetOperationFromContext for values of `set_operation`. If `validate_indices` is `True`, `SparseToSparseSetOperation` validates the order and range of `set1` and `set2` indices. Input `set1` is a `SparseTensor` represented by `set1_indices`, `set1_values`, and `set1_shape`. For `set1` ranked `n`, 1st `n-1` dimensions must be the same as `set2`. Dimension `n` contains values in a set, duplicates are allowed but ignored. Input `set2` is a `SparseTensor` represented by `set2_indices`, `set2_values`, and `set2_shape`. For `set2` ranked `n`, 1st `n-1` dimensions must be the same as `set1`. Dimension `n` contains values in a set, duplicates are allowed but ignored. If `validate_indices` is `True`, this op validates the order and range of `set1` and `set2` indices. Output `result` is a `SparseTensor` represented by `result_indices`, `result_values`, and `result_shape`. For `set1` and `set2` ranked `n`, this has rank `n` and the same 1st `n-1` dimensions as `set1` and `set2`. The `nth` dimension contains the result of `set_operation` applied to the corresponding `[0...n-1]` dimension of `set`. Args: set1_indices: A `Tensor` of type `int64`. 2D `Tensor`, indices of a `SparseTensor`. Must be in row-major order. set1_values: A `Tensor`. Must be one of the following types: `int8`, `int16`, `int32`, `int64`, `uint8`, `uint16`, `string`. 1D `Tensor`, values of a `SparseTensor`. Must be in row-major order. set1_shape: A `Tensor` of type `int64`. 1D `Tensor`, shape of a `SparseTensor`. `set1_shape[0...n-1]` must be the same as `set2_shape[0...n-1]`, `set1_shape[n]` is the max set size across `0...n-1` dimensions. set2_indices: A `Tensor` of type `int64`. 2D `Tensor`, indices of a `SparseTensor`. Must be in row-major order. set2_values: A `Tensor`. Must have the same type as `set1_values`. 1D `Tensor`, values of a `SparseTensor`. Must be in row-major order. set2_shape: A `Tensor` of type `int64`. 1D `Tensor`, shape of a `SparseTensor`. `set2_shape[0...n-1]` must be the same as `set1_shape[0...n-1]`, `set2_shape[n]` is the max set size across `0...n-1` dimensions. set_operation: A `string`. validate_indices: An optional `bool`. Defaults to `True`. name: A name for the operation (optional). Returns: A tuple of `Tensor` objects (result_indices, result_values, result_shape). result_indices: A `Tensor` of type `int64`. result_values: A `Tensor`. Has the same type as `set1_values`. result_shape: A `Tensor` of type `int64`. r^rrNrT) r`rarbrNrOrPrrrr)rrrrrr!_SparseToSparseSetOperationOutputrr r!r"r#r$-sparse_to_sparse_set_operation_eager_fallbackr&r'r(r)r*r+r,r-r.r/r0r1)r`rarbrNrOrPrrrr2r3r4r5r6r7r8r9r:r;r;r<sparse_to_sparse_set_operationJsj;          rez"raw_ops.SparseToSparseSetOperationc  Cst|d}|dur d}t|d}t||g| tjtjtjtjtj tj tj g\} } | \}}t |tj}t |tj}t |tj}t |tj}||||||g} d|d|d| f} tjdd| | | |d}trrtd| | |t|}|S) NrTrrsSparseToSparseSetOperationr>r?r^)r'r(r)rArBrCrDrErFrGrHrIr"rTrr,r1rcr)r`rarbrNrOrPrrrrrJrKr:r9r4r;r;r<rds0  2 rd)TN)@__doc__ collectionsZtensorflow.pythonrZtensorflow.python.eagerrrrr rr'Ztensorflow.python.frameworkrrBZtensorflow.security.fuzzing.pyrZ_atypesrZ_op_def_registryr r"r r*Z"tensorflow.python.util.deprecationr Ztensorflow.python.utilr _dispatchZ tensorflow.python.util.tf_exportr typingrr namedtuplerZInt16ZInt32ZInt64ZInt8StringZUInt16ZUInt8rZTensorFuzzingAnnotationstrboolr=Z to_raw_oprr%rQrMrSrLrRrUr\rZr[rcr_rer^rdr;r;r;r<sV            $(E&$@X>$><<$XjZ