diff options
author | 2016-09-07 15:44:17 -0800 | |
---|---|---|
committer | 2016-09-07 16:47:39 -0700 | |
commit | 9abc37359df75f8f45e4731a6317c677327c0c75 (patch) | |
tree | a127eb5a1e27cd63fae687cb8684be49733fd77f /tensorflow/python/ops/script_ops.py | |
parent | 4f09e634703d48a1487cf8c13fd3b0dc5c971e1e (diff) |
Switch to C++ shape inference fns for more ops in image_grad, io_ops,
script_ops, sparse_ops, and summary_ops.
Change: 132496531
Diffstat (limited to 'tensorflow/python/ops/script_ops.py')
-rw-r--r-- | tensorflow/python/ops/script_ops.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/ops/script_ops.py b/tensorflow/python/ops/script_ops.py index 69b524a5a9..0110b49b68 100644 --- a/tensorflow/python/ops/script_ops.py +++ b/tensorflow/python/ops/script_ops.py @@ -172,8 +172,8 @@ def py_func(func, inp, Tout, stateful=True, name=None): # pylint: enable=protected-access -ops.RegisterShape("PyFunc")(common_shapes.unknown_shape) -ops.RegisterShape("PyFuncStateless")(common_shapes.unknown_shape) +ops.RegisterShape("PyFunc")(common_shapes.call_cpp_shape_fn) +ops.RegisterShape("PyFuncStateless")(common_shapes.call_cpp_shape_fn) ops.NoGradient("PyFunc") ops.NoGradient("PyFuncStateless") |