From 52c3f583aadf836477935d7f95f82d73b71bdb40 Mon Sep 17 00:00:00 2001 From: Shivani Agrawal Date: Thu, 25 Oct 2018 11:12:27 -0700 Subject: [PATCH] Removes unnecessary unique id for defun (unlike Defun), defun can take non-unique function name for a process. PiperOrigin-RevId: 218717914 --- tensorflow/python/data/ops/dataset_ops.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tensorflow/python/data/ops/dataset_ops.py b/tensorflow/python/data/ops/dataset_ops.py index 6d07cc3e907..c8c1ab333b3 100644 --- a/tensorflow/python/data/ops/dataset_ops.py +++ b/tensorflow/python/data/ops/dataset_ops.py @@ -1808,8 +1808,7 @@ class StructuredFunctionWrapper(object): ".", "_")[:-2] if len(transformation_name) > 2 else "" self._func_name = "_".join([ readable_transformation_name, - function_utils.get_func_name(func), - str(ops.uid()) + function_utils.get_func_name(func) ]) # TODO(b/110122868): Enable this support for all `tf.data` functions.