
    Αi                         S r S rS rS rg)lookup_tablec                 N   U R                  5       R                  n/ nU R                  5       R                   He  nUR                  [
        :X  d  M  XR                  S5      S   :X  d  M2  UR                  UR                  S5       Vs/ s H  oRU   PM	     sn5        Mg     U$ s  snf )z
Find input variable of distribute lookup table in program.
We only support one distribute table now.
Args:
program(Program): given program, locate distributed lookup table
table_name(str): given table name that is found beforehand
Returns:
inputs
W    Ids)current_blockvarsglobal_blockopstypeLOOKUP_TABLE_TYPEinputextend)program
table_name
local_varsinputsopnames         j/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddle/distributed/distribute_lookup_table.py$find_distributed_lookup_table_inputsr      s     &&(--JF""$((77''XXc]1--BHHUOLOD$/OLM ) M M   B"
c                 N   U R                  5       R                  n/ nU R                  5       R                   He  nUR                  [
        :X  d  M  XR                  S5      S   :X  d  M2  UR                  UR                  S5       Vs/ s H  oRU   PM	     sn5        Mg     U$ s  snf )z
Find output variable of distribute lookup table in program.
We only support one distribute table now.
Args:
program(Program): given program, locate distributed lookup table
table_name(str): given table name that is found beforehand
Returns:
outputs
r   r   Out)	r   r   r	   r
   r   r   r   r   output)r   r   r   outputsr   r   s         r   %find_distributed_lookup_table_outputsr   %   s     &&(--JG""$((77''XXc]1--RYYu=MN=MT4 0=MNO ) N  Or   c                 T   SnU R                  5       R                   H  nUR                  [        :X  d  M  UR	                  S5      SL a;  Uc  UR                  S5      S   nXR                  S5      S   :w  a  [        S5      eMh  Uc  Mm  UR                  S5      S   U:w  a  M   e   U$ )z
Find distribute lookup table in program.
We only support one distribute table now.
Args:
program(Program): given program, locate distributed lookup table
Returns:
table_name or None
Nis_distributedTr   r   z;all distributed lookup_table_ops should have only one table)r	   r
   r   r   attrr   RuntimeError)r   r   r   s      r   find_distributed_lookup_tabler!   8   s     J""$((77''ww'(D0%!##q!1J#q!11&6  2 )88C=+z999 )     N)r   r   r   r!    r"   r   <module>r$      s    # &&r"   