[OpenCL] REGISTER -> REGISTER6

This commit is contained in:
Luke Iwanski 2017-06-22 13:00:38 +01:00
parent af0cbace1d
commit e911c74800

View File

@ -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