B +²ô`Mã@sXdZddlmZddlmZddlmZddlZddlZddlmZGdd„de ƒZ dS) z@Visitor restricting traversal to only the public tensorflow API.é)Úabsolute_import)Údivision)Úprint_functionN)Ú tf_inspectc@sReZdZdZdd„Zedd„ƒZedd„ƒZdd „Zdd d „Z d d„Z dd„Z d S)ÚPublicAPIVisitorzBVisitor to use with `traverse` to visit exactly the public TF API.cCsD||_d|_dddgdgdœ|_ddd d d d d gdgdgdœ|_dS)zàConstructor. `visitor` should be a callable suitable as a visitor for `traverse`. It will be called only for members of the public TensorFlow API. Args: visitor: A visitor to call for the public API. ÚtfÚcompilerÚcoreÚpythonZ cpp_flags)rztf.flagsZexamplesÚflagsÚplatformZpywrap_tensorflowZuser_opsZtoolsZ tensorboardZmock)rztf.appztf.testN)Ú_visitorÚ _root_nameÚ _private_mapÚ_do_not_descend_map)ÚselfZvisitor©rúQ/home/dcms/DCMS/lib/python3.7/site-packages/tensorflow/tools/common/public_api.pyÚ__init__ s  zPublicAPIVisitor.__init__cCs|jS)zÙA map from parents to symbols that should not be included at all. This map can be edited, but it should not be edited once traversal has begun. Returns: The map marking symbols to not include. )r)rrrrÚ private_mapPs zPublicAPIVisitor.private_mapcCs|jS)zØA map from parents to symbols that should not be descended into. This map can be edited, but it should not be edited once traversal has begun. Returns: The map marking symbols to not explore. )r)rrrrÚdo_not_descend_map\s z#PublicAPIVisitor.do_not_descend_mapcCs ||_dS)z'Override the default root name of 'tf'.N)r)rZ root_namerrrÚ set_root_namehszPublicAPIVisitor.set_root_nameNcCsF~||jkr||j|kpDt |¡ d¡r>t dt |¡¡ pD|dkS)z!Return whether a name is private.Ú_z__.*__$)Ú__base__Ú __class__)rÚsixÚ ensure_strÚ startswithÚreÚmatch)rÚpathÚnameÚobjrrrÚ _is_privatels zPublicAPIVisitor._is_privatecCs||jko||j|kS)zESafely queries if a specific fully qualified name should be excluded.)r)rr r!rrrÚ_do_not_descendus z PublicAPIVisitor._do_not_descendcCsÂt |¡r4tt |¡ d¡ƒdkr4td|j|fƒ‚|rHd |j|g¡n|j}x0t |ƒD]$\}}|  |||¡rX|  ||f¡qXW|  |||¡x.t |ƒD]"\}}|  ||¡r˜|  ||f¡q˜WdS)z.Visitor interface, see `traverse` for details.Ú.é zZModules nested too deep: %s.%s This is likely a problem with an accidental public import.N)rÚismoduleÚlenrrÚsplitÚ RuntimeErrorrÚjoinÚlistr#Úremover r$)rr ÚparentÚchildrenÚ full_pathr!ÚchildrrrÚ__call__zs  zPublicAPIVisitor.__call__)N) Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚpropertyrrrr#r$r2rrrrrs0  r) r6Ú __future__rrrrrZtensorflow.python.utilrÚobjectrrrrrÚs