U
    9%e                     @   s   d dl Zdd Zdd ZdS )    Nc                 C   s   t jj| S )zReturn a text dump of the LTC IR graph in dot format for the tensors.
    The text can be processed by tools like dot to be rendered in pdf,png etc.)torch_C_lazyZ_get_tensors_dot)tensors r   P/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/torch/_lazy/debug.pyrender_ir_graph   s    r   c                 C   s>   |dkrt jj| S |dkr,t jj| S td| dS )zReturn a dump of the tensors in the specified format.
    Valid format are
    - text: for LTC IR
    - backend: for the activate backend IR
    textbackendzUnrecognized IR format: N)r   r   r   Z_get_tensors_textZ_get_tensors_backendRuntimeError)r   Z	ir_formatr   r   r   dump_ir
   s
    r   )Ztorch._C._lazyr   r   r   r   r   r   r   <module>   s   