o Me8 ã@s dZdd„Zdd„Zdd„ZdS)Z lookup_tablecs\| ¡j‰g}| ¡jD]}|jtkr+|| d¡dkr+| ‡fdd„| d¡Dƒ¡q |S)a 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écóg|]}ˆ|‘qS©r©Ú.0Úname©Ú local_varsrúTD:\Projects\ConvertPro\env\Lib\site-packages\paddle/fluid/distribute_lookup_table.pyÚ !óz8find_distributed_lookup_table_inputs..ZIds)Ú current_blockÚvarsÚ global_blockÚopsÚtypeÚLOOKUP_TABLE_TYPEÚinputÚextend)ÚprogramÚ table_nameZinputsÚoprrr Ú$find_distributed_lookup_table_inputsó  €rcs\| ¡j‰g}| ¡jD]}|jtkr+|| d¡dkr+| ‡fdd„| d¡Dƒ¡q |S)a 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 rrcrrrrrrr r 4r z9find_distributed_lookup_table_outputs..ZOut) r rrrrrrrÚoutput)rrZoutputsrrrr Ú%find_distributed_lookup_table_outputs%rrcCs€d}| ¡jD]6}|jtkr=| d¡dur.|dur | d¡d}|| d¡dkr-tdƒ‚q|dur=| d¡d|ks=J‚q|S)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 NZis_distributedTrrz;all distributed lookup_table_ops should have only one table)rrrrÚattrrÚ RuntimeError)rrrrrr Úfind_distributed_lookup_table8s  ÿ€rN)rrrrrrrr Ús