o ®Ÿ?e« ã@sBdZddlZddlmZdgZejgZdd„Zdd„Zd d „Z dS) z5Helpers to traverse the Dataset dependency structure.éN)ÚdtypesZDummyIterationCountercCszg}t ¡}| |jj¡g}| ¡s;| ¡}| |¡||ƒr%| |¡|jD]}|j}||vr6| |¡q(| ¡r|S)zBTraverse a dataset graph, returning nodes matching `op_filter_fn`.) ÚqueueÚQueueÚputZ_variant_tensorÚopÚemptyÚgetÚappendZinputs)ÚdatasetZ op_filter_fnÚresultZbfs_qÚvisitedrÚiZinput_op©rúe/home/www/facesmatcher.com/pyenv/lib/python3.10/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|jdjtvp |jtvS©Nr)ÚoutputsÚdtypeÚTENSOR_TYPES_ALLOWLISTÚtypeÚOP_TYPES_ALLOWLIST©rrrrÚcapture_by_value<sÿz5obtain_capture_by_value_ops..capture_by_value©r)r rrrrÚobtain_capture_by_value_ops.s rcCst|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|jdjtjkSr)rrrÚvariantrrrrÚQsz/obtain_all_variant_tensor_ops..r)r rrrÚobtain_all_variant_tensor_opsCsr) Ú__doc__rZtensorflow.python.frameworkrrrrrrrrrrrÚs