Commit 6b1208c6 authored by Yong Tang's avatar Yong Tang Committed by Jonathan Hseu
Browse files

Add uint32 and uint64 kernel support for `Invert` (#15154)



* Add uint32 and uint64 kernel support for `Invert`

This fix adds uint32 and uint64 kernel support for `Invert`.

In bitwise_ops.cc, uint32 and uint64 have been registered
for `Invert` like other bitwise ops
`BitwiseAnd`/`BitwiseOr`/`BitwiseXor`/`LeftShift`/`RightShift`.
However, no uint32 and uint64 kernels available for `Invert` yet.

This fix add uint32 and uint64 kernel for `Invert`,
and adds additional test cases to cover the changes.

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Add test cases for uint32 and uint64 support with `Invert`

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Add missing uint32 and uint64 in GPU for invert

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Add DEFINE_UNARY8 for invert

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>
parent b1f5f433
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment