o
    biv                     @   s,   d dl mZ d dl mZ dd Zdd ZdS )    )backend)treec                    s`   ddl m  | stdtd tj fddg| R  W  d    S 1 s)w   Y  d S )Nr   ops)Cannot batch 0 values. Please file a bug.cpuc                     s
     | S )N)stackxsr    O/home/ubuntu/.local/lib/python3.10/site-packages/keras/src/utils/grain_utils.py<lambda>   s   
 zmake_batch.<locals>.<lambda>	keras.srcr   
ValueErrorr   device_scoper   map_structure)valuesr   r   r   
make_batch   s   $r   c                    sd   ddl m  | std fdd}td tj|g| R  W  d    S 1 s+w   Y  d S )Nr   r   r   c                     sL   t | d tr!t dkrdd l  fdd| D }  | } | S | S )Nr   
tensorflowc                    s   g | ]
} j | jd qS ))dtype)convert_to_tensorstring).0xtfr   r   
<listcomp>   s    z7make_string_batch.<locals>.batch_fn.<locals>.<listcomp>)
isinstancestrr   r   r   r	   r   r   r   batch_fn   s   

z#make_string_batch.<locals>.batch_fnr   r   )r   r    r   r   r   make_string_batch   s   $r!   N)r   r   r   r   r!   r   r   r   r   <module>   s    
