diff options
Diffstat (limited to 'tensorflow/python/framework/function_test.py')
| -rw-r--r-- | tensorflow/python/framework/function_test.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/tensorflow/python/framework/function_test.py b/tensorflow/python/framework/function_test.py index 65ca801cbe..83d256fab6 100644 --- a/tensorflow/python/framework/function_test.py +++ b/tensorflow/python/framework/function_test.py @@ -1227,6 +1227,15 @@ class FunctionsFromProtos(test.TestCase): ValueError, "FunctionDefLibrary contains cyclic gradient functions!"): function._from_library(library) + def testExperimentalAttrs(self): + + @function.Defun(dtypes.int32, experimental_tag="tag_value") + def FunctionWithAttr(i): + return array_ops.identity(i) + self.assertTrue("experimental_tag" in FunctionWithAttr.definition.attr) + self.assertEqual( + FunctionWithAttr.definition.attr["experimental_tag"].s, b"tag_value") + @test_util.with_c_api class FunctionOverloadTest(test.TestCase): |