o
    
i                  	   @   sn   d dl mZmZ d dlmZ d dlmZ d dlmZ d dl	m
Z
 	ddedee d	eee
 ef fd
dZdS )    )OptionalList)Floats2d)Model)registry)DocNtok2vecnOreturnc              	   C   s   t dd}t dd}t||dR | d}t dd}|||d}t dd}t dd	}t dd
}	t dd}
| | ? | ? |	|
||ddd? |? }|d| |d|  W d    |S 1 siw   Y  |S )Nlayerszchain.v1zclone.v1)z>>z**r	   z	Linear.v1)r	   nIzlist2ragged.v1zreduce_mean.v1zresidual.v1z	Maxout.v1   g        )r	   r   nPdropoutoutput_layerr   )r   getr   define_operatorsmaybe_get_dimset_ref)r   r	   chainclonetoken_widthLinearr   list2raggedreduce_meanresidualMaxoutmodel r   \/home/ubuntu/.local/lib/python3.10/site-packages/spacy_legacy/architectures/entity_linker.pyEntityLinker_v1	   s4   

r    )N)typingr   r   thinc.typesr   	thinc.apir   
spacy.utilr   spacy.tokensr   intr    r   r   r   r   <module>   s    