
    BVh                        d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ	 ddl
mZ dZd	Z ed
      j                  ed        ed      j                  ed        ed      d        Z ed      d        Z G d dej$                        Zy)z*Cardinality analysis of `Dataset` objects.    )dataset_ops)dtypes)ops)gen_dataset_ops)gen_experimental_dataset_ops)	tf_exportz&data.experimental.INFINITE_CARDINALITYINFINITEz%data.experimental.UNKNOWN_CARDINALITYUNKNOWNzdata.experimental.cardinalityc                 @    t        j                  | j                        S )a  Returns the cardinality of `dataset`, if known.

  The operation returns the cardinality of `dataset`. The operation may return
  `tf.data.experimental.INFINITE_CARDINALITY` if `dataset` contains an infinite
  number of elements or `tf.data.experimental.UNKNOWN_CARDINALITY` if the
  analysis fails to determine the number of elements in `dataset` (e.g. when the
  dataset source is a file).

  >>> dataset = tf.data.Dataset.range(42)
  >>> print(tf.data.experimental.cardinality(dataset).numpy())
  42
  >>> dataset = dataset.repeat()
  >>> cardinality = tf.data.experimental.cardinality(dataset)
  >>> print((cardinality == tf.data.experimental.INFINITE_CARDINALITY).numpy())
  True
  >>> dataset = dataset.filter(lambda x: True)
  >>> cardinality = tf.data.experimental.cardinality(dataset)
  >>> print((cardinality == tf.data.experimental.UNKNOWN_CARDINALITY).numpy())
  True

  Args:
    dataset: A `tf.data.Dataset` for which to determine cardinality.

  Returns:
    A scalar `tf.int64` `Tensor` representing the cardinality of `dataset`. If
    the cardinality is infinite or unknown, the operation returns the named
    constant `INFINITE_CARDINALITY` and `UNKNOWN_CARDINALITY` respectively.
  )r   dataset_cardinality_variant_tensor)datasets    c/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/data/experimental/ops/cardinality.pycardinalityr   !   s    > 
	,	,W-D-D	EE    z$data.experimental.assert_cardinalityc                       fd}|S )aT  Asserts the cardinality of the input dataset.

  NOTE: The following assumes that "examples.tfrecord" contains 42 records.

  >>> dataset = tf.data.TFRecordDataset("examples.tfrecord")
  >>> cardinality = tf.data.experimental.cardinality(dataset)
  >>> print((cardinality == tf.data.experimental.UNKNOWN_CARDINALITY).numpy())
  True
  >>> dataset = dataset.apply(tf.data.experimental.assert_cardinality(42))
  >>> print(tf.data.experimental.cardinality(dataset).numpy())
  42

  Args:
    expected_cardinality: The expected cardinality of the input dataset.

  Returns:
    A `Dataset` transformation function, which can be passed to
    `tf.data.Dataset.apply`.

  Raises:
    FailedPreconditionError: The assertion is checked at runtime (when iterating
      the dataset) and an error is raised if the actual and expected cardinality
      differ.
  c                     t        |       S )N)_AssertCardinalityDataset)r   expected_cardinalitys    r   	_apply_fnz%assert_cardinality.<locals>._apply_fn]   s    $W.BCCr    )r   r   s   ` r   assert_cardinalityr   C   s    4D 
r   c                   "     e Zd ZdZ fdZ xZS )r   z5A `Dataset` that assert the cardinality of its input.c                    || _         t        j                  |t        j                  d      | _        t        j                  | j                   j                  | j
                  fi | j                  }t        t        | 3  ||       y )Nr   )dtypename)_input_datasetr   convert_to_tensorr   int64_expected_cardinalityged_opsassert_cardinality_datasetr   _flat_structuresuperr   __init__)selfinput_datasetr   variant_tensor	__class__s       r   r'   z"_AssertCardinalityDataset.__init__f   s}    'D!$!6!6FLL7M"OD 77++""  

 N 

#T3M4BDr   )__name__
__module____qualname____doc__r'   __classcell__)r+   s   @r   r   r   c   s    =D Dr   r   N)r/   tensorflow.python.data.opsr   tensorflow.python.frameworkr   r   tensorflow.python.opsr   r   r#    tensorflow.python.util.tf_exportr   r   r   export_constantr,   r   r   UnaryUnchangedStructureDatasetr   r   r   r   <module>r7      s    1 2 . + 1 I 6 
 	
2 3 C Cj 	
1 2 B Bi
 *+F ,FB 12 3>D J J Dr   