
    BVhQ                     h    d Z ddlmZ ddlmZ ddlmZ ddlmZ d Z G d dej                        Z
y	)
z.The implementation of `tf.data.Dataset.shard`.    )dataset_ops)dtypes)ops)gen_dataset_opsc                     t        | |||      S )"See `Dataset.shard()` for details.)_ShardDataset)input_dataset
num_shardsindexnames       S/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/data/ops/shard_op.py_shardr      s    	}j%	>>    c                   "     e Zd ZdZ fdZ xZS )r	   z#A `Dataset` for sharding its input.c                 p   || _         t        j                  |t        j                  d      | _        t        j                  |t        j                  d      | _        || _        t        j                  |j                  f| j
                  | j                  d| j                  }t        | 5  ||       y)r   r   )dtyper   r   )r   r   N)_input_datasetr   convert_to_tensorr   int64_num_shards_index_namer   shard_dataset_variant_tensor_common_argssuper__init__)selfr
   r   r   r   variant_tensor	__class__s         r   r   z_ShardDataset.__init__   s    'D,,&,,\;D''V\\PDKDJ$22%%##kk 

	N
 
G]N3r   )__name__
__module____qualname____doc__r   __classcell__)r!   s   @r   r	   r	      s    +4 4r   r	   N)r%   tensorflow.python.data.opsr   tensorflow.python.frameworkr   r   tensorflow.python.opsr   r   UnaryUnchangedStructureDatasetr	    r   r   <module>r,      s-    5 2 . + 1?
4K>> 4r   