[OpenCL] REGISTER -> REGISTER6
This commit is contained in:
parent
af0cbace1d
commit
e911c74800
@ -20,7 +20,7 @@ REGISTER6(UnaryOp, CPU, "Invert", functor::invert, int8, int16, int32, int64,
|
||||
uint8, uint16);
|
||||
|
||||
#ifdef TENSORFLOW_USE_SYCL
|
||||
REGISTER(UnaryOp, SYCL, "Invert", functor::invert, int8, int16, int32, int64,
|
||||
REGISTER6(UnaryOp, SYCL, "Invert", functor::invert, int8, int16, int32, int64,
|
||||
uint8, uint16);
|
||||
#endif // TENSORFLOW_USE_SYCL
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user