קgh~ddlZddlmZmZddlZddlmZd dZdZdej j deefdZ d de fd Z dS) N)DictOptional) LazyStringc fddvrdd<dvr#tjsdd<tfdS)z; Returns a LazyString that formats the graph code. cdSS)N )maybe_idnamesK/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/fx/_utils.py format_namez+lazy_format_graph_code..format_names"  ''X'' 'K print_outputFcoloredc rtddjjjjdiS)Nz===== z ===== r )_format_graph_codeforward__code__ co_filenameprint_readable)r gmkwargssr z(lazy_format_graph_code..sG" ,[[]] , , , J  + B  ' ' ' '  r)sysstdoutisattyr)r rr rr s````@r lazy_format_graph_coder s  V##!&~F3:#4#4#6#6!y         rcd|d|d|dS)z7 Returns a string that formats the graph code. zTRACED GRAPH r r )r filename graph_strs r rr#s% =T < d|dddS)zD Returns a string of the last num_nodes nodes in the graph. rootrN)rangeappend format_noder(prevjoin)r* num_nodes node_contextscuris r get_node_contextr94s{M C 9  S__../// 6V   Eh 99]44R4( ) ))r)N)r,)rtypingrrtorchtorch._loggingrrrfxGraphr+strr9r rr r@s !!!!!!!! %%%%%%4===ux~(4. * *3 * * * * * *r