o #i@sdddlmZmZddlZddlmZdddZddZdejj d eefd d Z dd e fd dZ dS))DictOptionalN LazyStringc s2fdddvrdd<tfddS)z; Returns a LazyString that formats the graph code. csdur dSS)N r)maybe_idnamerM/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/torch/fx/_utils.py format_namesz+lazy_format_graph_code..format_name print_outputFcs(tddjjjjdiS)Nz===== z ===== r)_format_graph_codeforward__code__ co_filenameprint_readabler)r gmkwargsrr s  z(lazy_format_graph_code..r)r rrrr)r rrrr r lazy_format_graph_code s rcCsd|d|d|dS)z7 Returns a string that formats the graph code. zTRACED GRAPH r r)r filename graph_strrrr r sr graphreturncCs2|jD]}|jdkrd|jvr|jdSqdS)zF Returns the nn_module_stack of the first call_function node. call_functionnn_module_stackN)nodesopmeta)rnoderrr #first_call_function_nn_module_stack's r!cCsLg}|}t|D]}|||jdkrn|j}qd|dddS)zD Returns a string of the last num_nodes nodes in the graph. rootrN)rangeappend format_noderprevjoin)r num_nodes node_contextscurirrr get_node_context1s  r.)N)r") typingrrtorchtorch._loggingrrr fxGraphr!strr.rrrr s