diff options
author | 2016-11-14 10:36:07 -0800 | |
---|---|---|
committer | 2016-11-14 10:44:25 -0800 | |
commit | 897c3e2dd650b71d96bd0886cea17474a2d229fb (patch) | |
tree | af610cd3906c3204655113df2df67843afd83544 /tensorflow/python/ops/script_ops.py | |
parent | baa64c39edc0854e9ba5ca1e3d16ac633de6056e (diff) |
Make the python default for shape fn be call_cpp_shape_fn.
Remove explicit RegisterShape calls that used call_cpp_shape_fn without
input_tensors_needed arguments.
Remove requirement that C++ shape functions be specified.
Change: 139089667
Diffstat (limited to 'tensorflow/python/ops/script_ops.py')
-rw-r--r-- | tensorflow/python/ops/script_ops.py | 4 |
1 files changed, 0 insertions, 4 deletions
diff --git a/tensorflow/python/ops/script_ops.py b/tensorflow/python/ops/script_ops.py index 5ee6cb85fb..b89b76cf55 100644 --- a/tensorflow/python/ops/script_ops.py +++ b/tensorflow/python/ops/script_ops.py @@ -31,7 +31,6 @@ import threading import numpy as np from tensorflow.python import pywrap_tensorflow -from tensorflow.python.framework import common_shapes from tensorflow.python.framework import ops from tensorflow.python.ops import gen_script_ops @@ -199,8 +198,5 @@ def py_func(func, inp, Tout, stateful=True, name=None): return result if is_list_or_tuple else result[0] -ops.RegisterShape("PyFunc")(common_shapes.call_cpp_shape_fn) -ops.RegisterShape("PyFuncStateless")(common_shapes.call_cpp_shape_fn) - ops.NotDifferentiable("PyFunc") ops.NotDifferentiable("PyFuncStateless") |