diff options
author | 2017-09-05 10:02:11 -0700 | |
---|---|---|
committer | 2017-09-05 10:09:53 -0700 | |
commit | 9eb8dd7d64d36f6ae0b268098597994bb7e6ffbc (patch) | |
tree | f0f8ffaedd54df0d9834b978292d3beee7303831 /tensorflow/contrib/stateless | |
parent | 370c9b8ae787cf56cf5bc5586876fac50867edb4 (diff) |
Mark tf.contrib.stateless.* ops as NotDifferentiable
PiperOrigin-RevId: 167592965
Diffstat (limited to 'tensorflow/contrib/stateless')
-rw-r--r-- | tensorflow/contrib/stateless/BUILD | 1 | ||||
-rw-r--r-- | tensorflow/contrib/stateless/__init__.py | 6 |
2 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/contrib/stateless/BUILD b/tensorflow/contrib/stateless/BUILD index 598e6513ae..865fb72a55 100644 --- a/tensorflow/contrib/stateless/BUILD +++ b/tensorflow/contrib/stateless/BUILD @@ -21,6 +21,7 @@ py_library( srcs_version = "PY2AND3", deps = [ ":stateless_random_ops", + "//tensorflow/python:framework", "//tensorflow/python:util", ], ) diff --git a/tensorflow/contrib/stateless/__init__.py b/tensorflow/contrib/stateless/__init__.py index 82e5d36ce4..ca937546f5 100644 --- a/tensorflow/contrib/stateless/__init__.py +++ b/tensorflow/contrib/stateless/__init__.py @@ -34,5 +34,11 @@ from __future__ import print_function # pylint: disable=wildcard-import from tensorflow.contrib.stateless.gen_stateless_random_ops import * +from tensorflow.python.framework import ops from tensorflow.python.util.all_util import remove_undocumented + +ops.NotDifferentiable("StatelessRandomNormal") +ops.NotDifferentiable("StatelessRandomUniform") +ops.NotDifferentiable("StatelessTruncatedNormal") + remove_undocumented(__name__) |