U yh@sbddlmZmZddlZddlmZdddZddZejj eedd d Z de d d dZ dS))DictOptionalN LazyStringc s2fdddkrdd<tfddS)z; Returns a LazyString that formats the graph code. csdk rdSSdS)N r)maybe_idnamerA/var/www/html/venv/lib/python3.8/site-packages/torch/fx/_utils.py format_namesz+lazy_format_graph_code..format_nameZ print_outputFcs$tddjjjjfS)Nz===== z ===== )_format_graph_codeforward__code__ co_filenameZprint_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 filenameZ graph_strrrr r sr )graphreturncCs2|jD]&}|jdkrd|jkr|jdSqdS)zF Returns the nn_module_stack of the first call_function node. Z call_functionZnn_module_stackN)Znodesopmeta)rnoderrr #first_call_function_nn_module_stack's r)rcCsLg}|}t|D]&}|||jdkr0q8|j}qd|dddS)zD Returns a string of the last num_nodes nodes in the graph. rootrN)rangeappendZ format_noderprevjoin)rZ num_nodesZ node_contextscurirrr get_node_context1s  r%)N)r) typingrrZtorchZtorch._loggingrrr ZfxZGraphrstrr%rrrr s