diff options
author | Yong Tang <yong.tang.github@outlook.com> | 2017-12-26 20:49:55 -0600 |
---|---|---|
committer | drpngx <drpngx@users.noreply.github.com> | 2017-12-26 18:49:55 -0800 |
commit | 489710701b1cded21ed4bf67e556aceea3bce340 (patch) | |
tree | dab44e453c34ebbb6b610a445b16f6767b8261eb /ADOPTERS.md | |
parent | e34ea8ae8c27bdeafacb12d40aeb008ace6a981a (diff) |
Fix discrepancy between doc and impl for `tf.reverse` (#13672)
* Fix discrepancy between doc and impl for `tf.reverse`
This fix tries to address the discrepancy between doc and
implementations for `tf.reverse`. In the documentation,
both int32 and int64 could be used for axis.
However, the actual implementation of the `tf.reverse` does
not support int64 and caused missing kernel error:
```python
ubuntu@ubuntu:~$ python
Python 2.7.12 (default, Nov 19 2016, 06:48:10)
[GCC 5.4.0 20160609] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
>>> v = tf.reverse_v2([1, 2, 3], tf.constant([0], tf.int64))
>>> sess = tf.Session()
>>> sess.run(v)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py", line 889, in run
run_metadata_ptr)
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py", line 1120, in _run
feed_dict_tensor, options, run_metadata)
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py", line 1317, in _do_run
options, run_metadata)
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/client/session.py", line 1336, in _do_call
raise type(e)(node_def, op, message)
tensorflow.python.framework.errors_impl.InvalidArgumentError: No OpKernel was registered to support Op 'ReverseV2' with these attrs. Registered devices: [CPU], Registered kernels:
device='CPU'; T in [DT_STRING]; Tidx in [DT_INT32]
device='CPU'; T in [DT_BOOL]; Tidx in [DT_INT32]
device='CPU'; T in [DT_COMPLEX128]; Tidx in [DT_INT32]
device='CPU'; T in [DT_COMPLEX64]; Tidx in [DT_INT32]
device='CPU'; T in [DT_DOUBLE]; Tidx in [DT_INT32]
device='CPU'; T in [DT_FLOAT]; Tidx in [DT_INT32]
device='CPU'; T in [DT_HALF]; Tidx in [DT_INT32]
device='CPU'; T in [DT_INT8]; Tidx in [DT_INT32]
device='CPU'; T in [DT_UINT8]; Tidx in [DT_INT32]
device='CPU'; T in [DT_INT16]; Tidx in [DT_INT32]
device='CPU'; T in [DT_UINT16]; Tidx in [DT_INT32]
device='CPU'; T in [DT_INT32]; Tidx in [DT_INT32]
device='CPU'; T in [DT_INT64]; Tidx in [DT_INT32]
[[Node: ReverseV2 = ReverseV2[T=DT_INT32, Tidx=DT_INT64](ReverseV2/tensor, Const)]]
```
This fix addresses this issue and added the int64 support
for axis in `tf.reverse`.
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add GPU support of int64 axis in `tf.reverse`
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add test cases for int64 support of axis in `tf.reverse`
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add OpenCL int64 support of axis in `tf.reverse`
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
Diffstat (limited to 'ADOPTERS.md')
0 files changed, 0 insertions, 0 deletions