B Ù»ˆd« ã@sBdZddlZddlmZdgZejgZdd„Zdd„Zd d „Z dS) z5Helpers to traverse the Dataset dependency structure.éN)ÚdtypesZDummyIterationCountercCs|g}t ¡}| |jj¡g}xX| ¡sv| ¡}| |¡||ƒrL| |¡x&|jD]}|j}||krT| |¡qTWq W|S)zBTraverse a dataset graph, returning nodes matching `op_filter_fn`.) ÚqueueÚQueueÚputZ_variant_tensorÚopÚemptyÚgetÚappendÚinputs)ÚdatasetZ op_filter_fnÚresultZbfs_qÚvisitedrÚiZinput_op©rúV/var/www/html/venv/lib/python3.7/site-packages/tensorflow/python/data/util/traverse.pyÚ _traverses    rcCsdd„}t||ƒS)aHGiven an input dataset, finds all allowlisted ops used for construction. Allowlisted ops are stateful ops which are known to be safe to capture by value. Args: dataset: Dataset to find allowlisted stateful ops for. Returns: A list of variant_tensor producing dataset ops used to construct this dataset. cSs|jdjtkp|jtkS)Nr)ÚoutputsÚdtypeÚTENSOR_TYPES_ALLOWLISTÚtypeÚOP_TYPES_ALLOWLIST)rrrrÚcapture_by_value<sz5obtain_capture_by_value_ops..capture_by_value)r)r rrrrÚobtain_capture_by_value_ops.srcCst|dd„ƒS)a¯Given an input dataset, finds all dataset ops used for construction. A series of transformations would have created this dataset with each transformation including zero or more Dataset ops, each producing a dataset variant tensor. This method outputs all of them. Args: dataset: Dataset to find variant tensors for. Returns: A list of variant_tensor producing dataset ops used to construct this dataset. cSs|jdjtjkS)Nr)rrrÚvariant)rrrrÚQóz/obtain_all_variant_tensor_ops..)r)r rrrÚobtain_all_variant_tensor_opsCsr) Ú__doc__rZtensorflow.python.frameworkrrrrrrrrrrrÚs