B лˆdÀã@sÖdZddlZddlZddlZddlmZejZe dd„¡Z e dd„¡Z Gdd„de ƒZ Gd d „d e ƒZ iZGd d „d e ƒZGd d„de ƒZeƒZGdd„deƒZGdd„deƒZdd„Zddd„ZejZejZdS)zGFunctions used to extract and analyze stacks. Faster than Python libs.éN)Ú _tf_stackcCstƒgS)N)ÚSentinelMapper©rrúQ/var/www/html/venv/lib/python3.7/site-packages/tensorflow/python/util/tf_stack.pyÚ#órcCstƒgS)N)ÚSentinelFilterrrrrr$rc@s0eZdZdZdZdZdd„Zdd„Zdd„ZdS) ÚStackTraceTransformz4Base class for stack trace transformation functions.NcCsT|jdkrtƒ|_n|jtƒks(tdƒ‚|j|j}|d|_| |¡| ¡|S)NzShared across threads?éÿÿÿÿ)Ú _thread_keyÚ_get_thread_keyÚAssertionErrorÚ _stack_dictÚparentÚappendÚupdate)ÚselfÚstackrrrÚ __enter__-s     zStackTraceTransform.__enter__cCs$|j|j ¡}||ks tdƒ‚dS)NzConcurrent access?)rr Úpopr )rZ unused_typeZ unused_valueZunused_tracebackÚtoprrrÚ__exit__;szStackTraceTransform.__exit__cCs tdƒ‚dS)Nz subclasses need to override this)ÚNotImplementedError)rrrrr?szStackTraceTransform.update) Ú__name__Ú __module__Ú __qualname__Ú__doc__rr rrrrrrrr 's r c@s,eZdZdZeZdd„Zdd„Zdd„ZdS) ÚStackTraceMapperz@Allows remapping traceback information to different source code.cCst ¡|_dS)N)rZPyBindSourceMapÚ internal_map)rrrrÚ__init__GszStackTraceMapper.__init__cCs|j t| ¡ ¡ƒ¡dS)N)rÚ update_toÚtupleÚget_effective_source_mapÚitems)rrrrrJszStackTraceMapper.updatecCs tdƒ‚dS)zFReturns a map (filename, lineno) -> (filename, lineno, function_name).z subclasses need to override thisN)r)rrrrr"Msz)StackTraceMapper.get_effective_source_mapN) rrrrÚ_source_mapper_stacksrrrr"rrrrrCs rc@seZdZdd„ZdS)rcCstS)N)Ú EMPTY_DICT)rrrrr"Wsz'SentinelMapper.get_effective_source_mapN)rrrr"rrrrrUsrc@s,eZdZdZeZdd„Zdd„Zdd„ZdS) ÚStackTraceFilterzFAllows filtering traceback information by removing superfluous frames.cCst ¡|_dS)N)rZ PyBindFileSetÚ internal_set)rrrrr_szStackTraceFilter.__init__cCs|j t| ¡ƒ¡dS)N)r'r ÚsetÚget_filtered_filenames)rrrrrbszStackTraceFilter.updatecCs tdƒ‚dS)Nz subclasses need to override this)r)rrrrr)esz'StackTraceFilter.get_filtered_filenamesN) rrrrÚ_source_filter_stacksrrrr)rrrrr&[s r&c@seZdZdd„ZdS)rcCstS)N)Ú EMPTY_SET)rrrrr)nsz%SentinelFilter.get_filtered_filenamesN)rrrr)rrrrrlsrcs(eZdZdZ‡fdd„Zdd„Z‡ZS)ÚCurrentModuleFilterzFFilters stack frames from the module where this is used (best effort).csVtƒ ¡d}d}t ¡}z0|dk r<|j}|dk r|jdk r|jSt|jfƒ}|jdk r4||j ¡O}||_|S)N)r3Ú frozensetr2rr))rZfiltered_filenamesrrrr)‹s   z*CurrentModuleFilter.get_filtered_filenames)rrrrrr)Ú __classcell__rr)r5rr,rs r,cCs&tƒ}t t|djt|dj¡S)a#An eager-friendly alternative to traceback.extract_stack. Returns: A list-like FrameSummary containing StackFrame-like objects, which are namedtuple-like objects with the following fields: filename, lineno, name, line, meant to masquerade as traceback.FrameSummary objects. r )r rÚ extract_stackr$rr*r')Ú thread_keyrrrr8–s  r8écCs.tƒ}t t|djt|dj||¡dS)zàAttaches the current stack trace to `c_op`. Args: c_op: a TF_Operation object. stacklevel: An integer for ignoring Python wrapper stack frames. The default value of 1 ignores this function from the frame. r N)r rÚextract_stack_for_opr$rr*r')Zc_opÚ stacklevelr9rrrr;©s  r;)r:)rÚ collectionsr.Ú threadingÚtensorflow.python.utilrÚ get_identr Ú defaultdictr$r*Úobjectr rr%rr&r6r+rr,r8r;ZStackTraceWrapperÚ StackSummaryZ StackFrameÚ FrameSummaryrrrrÚs& $