o
    ?e{                     @   sR   d Z ddlmZ ddlmZ ddlmZ ddlmZ dd Z	G dd	 d	ej
Zd
S )z/The implementation of `tf.data.Dataset.unique`.    )dataset_ops)nest)dtypes)gen_experimental_dataset_opsc                 C   s
   t | |S N)_UniqueDataset)input_datasetname r
   e/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/data/ops/unique_op.py_unique   s   
r   c                       s"   e Zd ZdZd fdd	Z  ZS )r   z=A dataset containing the unique elements of an input dataset.Nc                    sp   || _ tt|D ]}|tjtjtjfvrt	d| dq|| _
tj| j jfi | j}t || dS )z)See `tf.data.Dataset.unique` for details.z/`tf.data.Dataset.unique` does not support type z? -- only `tf.int32`, `tf.int64`, and `tf.string` are supported.N)Z_input_datasetr   flattenr   Zget_legacy_output_typesr   Zint32Zint64string	TypeError_namer   Zunique_datasetZ_variant_tensorZ_common_argssuper__init__)selfr   r	   tyZvariant_tensor	__class__r
   r   r      s   
z_UniqueDataset.__init__r   )__name__
__module____qualname____doc__r   __classcell__r
   r
   r   r   r      s    r   N)r   Ztensorflow.python.data.opsr   Ztensorflow.python.data.utilr   Ztensorflow.python.frameworkr   Ztensorflow.python.opsr   r   ZUnaryUnchangedStructureDatasetr   r
   r
   r
   r   <module>   s   