j~ddlZddlmZmZddlZddlmZd dZdZdej j deefdZ d de fd Z dS) N)DictOptional) LazyStringc fddvrdd<dvr: tjsdd<n#t$rdd<YnwxYwt fdS)z; Returns a LazyString that formats the graph code. cdSS)N )maybe_idnames@C:\Projects\Historical\venv\Lib\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)sysstdoutisattyAttributeErrorr)r rr rr s````@r lazy_format_graph_coder s  V##!&~F &:$$&& *$)y! & & & %F9    &         s#>AAcd|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. rootr N)rangeappend format_noder)prevjoin)r+ num_nodes node_contextscur_s r get_node_contextr:8s{M C 9  S__../// 6V   Eh 99]44R4( ) ))r)N)r-)rtypingrrtorchtorch._loggingrrrfxGraphr,strr:r rr rAs !!!!!!!! %%%%%%<===ux~(4. * *3 * * * * * *r