commit 4d7c477213ac6c2b0af24bf2c79200070731512e
author Sergio Guadarrama <sguada@google.com> Tue Aug 22 16:54:41 2017 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Tue Aug 22 16:58:39 2017 -0700
tree 2c1b097a83995dac5dfe77c50960053c363a8d1c
parent 3ae80f7bf79a295e375ddbe3659d89b55ca6f0bd

Make sure the module information is preserved by the tf_decorator.

PiperOrigin-RevId: 166134795

tensorflow/python/util/tf_decorator.py

diff --git a/tensorflow/python/util/tf_decorator.py b/tensorflow/python/util/tf_decorator.py
index a5d979e..8b4d22f 100644
--- a/tensorflow/python/util/tf_decorator.py
+++ b/tensorflow/python/util/tf_decorator.py
@@ -88,6 +88,7 @@
                           decorator_argspec)
   setattr(decorator_func, '_tf_decorator', decorator)
   decorator_func.__name__ = target.__name__
+  decorator_func.__module__ = target.__module__
   decorator_func.__doc__ = decorator.__doc__
   decorator_func.__wrapped__ = target
   return decorator_func