BVh NdZddlZddlmZdgZej gZdZdZdZ y)z5Helpers to traverse the Dataset dependency structure.N)dtypesDummyIterationCountercg}tj}|j|jjg}|j s~|j }|j|||r|j||jD]$}|j}||vs|j|&|j s~|S)zBTraverse a dataset graph, returning nodes matching `op_filter_fn`.) queueQueueput_variant_tensoropemptygetappendinputs)dataset op_filter_fnresultbfs_qvisitedr iinput_ops T/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/data/util/traverse.py _traversers & ++-%))G # # & &' ' KKM B NN2B mmB YYh   ( KKM -c d}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. ch|jdjtvxs|jtvSNr)outputsdtypeTENSOR_TYPES_ALLOWLISTtypeOP_TYPES_ALLOWLISTr s rcapture_by_valuez5obtain_capture_by_value_ops..capture_by_value<s0 JJqM  #9 9 * GG) )+rr)rr"s robtain_capture_by_value_opsr$.s+ 7, --rct|dS)aGiven 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. cV|jdjtjk(Sr)rrrvariantr!s rz/obtain_all_variant_tensor_ops..Qsrzz!}':':fnn'Lrr#)rs robtain_all_variant_tensor_opsr)Cs 7L MMr) __doc__rtensorflow.python.frameworkrr r'rrr$r)rrr-s:< ...!..)$.*Nr