o i@s*dZddlZddlmZGdddZdS)z@Visitor restricting traversal to only the public tensorflow API.N) tf_inspectc@sReZdZdZddZeddZeddZdd Zdd d Z d dZ ddZ d S)PublicAPIVisitorzBVisitor to use with `traverse` to visit exactly the public TF API.cCs8||_d|_gddgd|_gddgdgd|_d S) zConstructor. `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)compilercoresecuritydtensorpythontsl cpp_flags)rztf.flags)examplesflagsplatformpywrap_tensorflowuser_opstools tensorboardr mock)rztf.appztf.testN)_visitor _root_name _private_map_do_not_descend_map)selfvisitorrX/mnt/sdb/aimis/docanh/lib/python3.10/site-packages/tensorflow/tools/common/public_api.py__init__s  zPublicAPIVisitor.__init__cC|jS)zA 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. )rrrrr private_mapT zPublicAPIVisitor.private_mapcCr)zA 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. rrrrrdo_not_descend_map`r z#PublicAPIVisitor.do_not_descend_mapcCs ||_dS)z'Override the default root name of 'tf'.N)r)r root_namerrr set_root_namels zPublicAPIVisitor.set_root_nameNcCs:~||jvr ||j|vp|drtd| p|dvS)z!Return whether a name is private._z__.*__$)__base__ __class____next_in_mro__)r startswithrematch)rpathnameobjrrr _is_privateps zPublicAPIVisitor._is_privatecCs||jvo ||j|vS)zESafely queries if a specific fully qualified name should be excluded.r!)rr,r-rrr_do_not_descendxs  z PublicAPIVisitor._do_not_descendcCst|rt|ddkrtd|j|f|r!d|j|gn|j}t|D]\}}||||r:| ||fq(| |||t|D]\}}| ||rW| ||fqFdS)z.Visitor interface, see `traverse` for details.. zZModules nested too deep: %s.%s This is likely a problem with an accidental public import.N) rismodulelensplit RuntimeErrorrjoinlistr/removerr0)rr,parentchildren full_pathr-childrrr__call__}s zPublicAPIVisitor.__call__)N) __name__ __module__ __qualname____doc__rpropertyrr"r$r/r0r>rrrrrs;    r)rBr*tensorflow.python.utilrrrrrrs