
    x-j                     X     G d  d          Z  G d de           Z G d de           ZdS )c                       e Zd Zd ZddZdS )_DatasetFetcherc                 >    || _         || _        || _        || _        d S N)datasetauto_collate_batch
collate_fn	drop_last)selfr   r   r   r	   s        \/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddle/io/dataloader/fetcher.py__init__z_DatasetFetcher.__init__   s#    "4$"    Nc                 :    t          d| j        j                   )Nz 'fetch' not implement for class )NotImplementedError	__class____name__)r
   batch_indices
done_events      r   fetchz_DatasetFetcher.fetch    s$    !Ht~/FHH
 
 	
r   r   )r   
__module____qualname__r   r    r   r   r   r      s7        # # #
 
 
 
 
 
r   r   c                   &     e Zd Z fdZddZ xZS )_IterableDatasetFetcherc                 x    t                                          ||||           t          |          | _        d S r   )superr   iterdataset_iterr
   r   r   r   r	   r   s        r   r   z _IterableDatasetFetcher.__init__'   s6    "4j)LLL MMr   Nc                    | j         rg }|D ]T}||                                s:	 |                    t          | j                             A# t
          $ r Y  nw xY w d S t          |          dk    s'| j        r't          |          t          |          k     rt
          nt          | j                  }| j        r|                     |          }|S )N    )	r   is_setappendnextr   StopIterationlenr	   r   )r
   r   r   data_s        r   r   z_IterableDatasetFetcher.fetch+   s    " 	+D"    %Z->->-@-@%D):$;$;<<<<(     444yyA~~ #&t99s=/A/A#A#A## )**D? 	)??4((Ds   'A
AAr   r   r   r   r   r   __classcell__r   s   @r   r   r   &   sL        * * * * *       r   r   c                   &     e Zd Z fdZddZ xZS )_MapDatasetFetcherc                 P    t                                          ||||           d S r   )r   r   r   s        r   r   z_MapDatasetFetcher.__init__E   s(    "4j)LLLLLr   Nc                     | j         rAg }|D ];}||                                s!|                    | j        |                    9 d S n| j        |         }| j        r|                     |          }|S r   )r   r!   r"   r   r   )r
   r   r   r&   idxs        r   r   z_MapDatasetFetcher.fetchH   s    " 		/D$    %Z->->-@-@%KKS 1222244	  <.D? 	)??4((Dr   r   r(   r*   s   @r   r,   r,   D   sQ        M M M M M       r   r,   N)r   r   r,   r   r   r   <module>r0      s    
 
 
 
 
 
 
 
,    o   <         r   