Some questions about RegisterShape

[alongwithyou]

Traceback (most recent call last):
File "tests/test_warprnnt_op.py", line 3, in
from warprnnt_tensorflow import rnnt_loss
File "/NEWAI/Speech/Member/gaoxinglong/bin/anaconda3/lib/python3.7/site-packages/warprnnt_tensorflow-0.1-py3.7-linux-x86_64.egg/warprnnt_tensorflow/init.py", line 44, in
@ops.RegisterShape("WarpRNNT")
AttributeError: module 'tensorflow.python.framework.ops' has no attribute 'RegisterShape'

[PeiyanFlying]

Hello sir, I meet with the same problem as yours. So have you solved this bug?
Really appreciate your answer.
Many thanks!

[bhaswa]

I am also getting the same issue. Any update on this?
Thanks!

[pyyush]

TFv2.3.0

======================================================================
ERROR: test_basic (unittest.loader._FailedTest)

ImportError: Failed to import test module: test_basic
Traceback (most recent call last):
File "/N/soft/rhel7/deeplearning/Python-3.8.5/lib/python3.8/unittest/loader.py", line 436, in _find_test_path
module = self._get_module_from_name(name)
File "/N/soft/rhel7/deeplearning/Python-3.8.5/lib/python3.8/unittest/loader.py", line 377, in _get_module_from_name
import(name)
File "/N/slate/piyush/warp-transducer/tensorflow_binding/tests/test_basic.py", line 3, in
from warprnnt_tensorflow import rnnt_loss
File "/N/slate/piyush/warp-transducer/tensorflow_binding/warprnnt_tensorflow/init.py", line 44, in
@ops.RegisterShape("WarpRNNT")
AttributeError: module 'tensorflow.python.framework.ops' has no attribute 'RegisterShape'

======================================================================
ERROR: test_warprnnt_op (unittest.loader._FailedTest)

ImportError: Failed to import test module: test_warprnnt_op
Traceback (most recent call last):
File "/N/soft/rhel7/deeplearning/Python-3.8.5/lib/python3.8/unittest/loader.py", line 436, in _find_test_path
module = self._get_module_from_name(name)
File "/N/soft/rhel7/deeplearning/Python-3.8.5/lib/python3.8/unittest/loader.py", line 377, in _get_module_from_name
import(name)
File "/N/slate/piyush/warp-transducer/tensorflow_binding/tests/test_warprnnt_op.py", line 3, in
from warprnnt_tensorflow import rnnt_loss
File "/N/slate/piyush/warp-transducer/tensorflow_binding/warprnnt_tensorflow/init.py", line 38, in
def _RNNTLossGrad(op, grad_loss, _):
File "/N/soft/rhel7/deeplearning/Python-3.8.5/lib/python3.8/site-packages/tensorflow/python/framework/ops.py", line 2598, in call
_gradient_registry.register(f, self._op_type)
File "/N/soft/rhel7/deeplearning/Python-3.8.5/lib/python3.8/site-packages/tensorflow/python/framework/registry.py", line 58, in register
raise KeyError(
KeyError: "Registering two gradient with name 'WarpRNNT'! (Previous registration was in register /N/soft/rhel7/deeplearning/Python-3.8.5/lib/python3.8/site-packages/tensorflow/python/framework/registry.py:66)"

---

Ran 2 tests in 0.000s

FAILED (errors=2)
Test failed: <unittest.runner.TextTestResult run=2 errors=2 failures=0>
error: Test failed: <unittest.runner.TextTestResult run=2 errors=2 failures=0>

[muyamu]

tensorflow2.0 version has RegisterShape.

[wilson1692]

I have a problem, can help me please
AttributeError: module 'tensorflow.python.framework.ops' has no attribute 'RegisterShape'