diff options
Diffstat (limited to 'tensorflow/core/kernels/cast_op.cc')
-rw-r--r-- | tensorflow/core/kernels/cast_op.cc | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/tensorflow/core/kernels/cast_op.cc b/tensorflow/core/kernels/cast_op.cc index 562934ed63..8bad488482 100644 --- a/tensorflow/core/kernels/cast_op.cc +++ b/tensorflow/core/kernels/cast_op.cc @@ -239,12 +239,11 @@ class SyclCastOp : public CastOpBase { }; #define REGISTER_CAST_SYCL(srctype, dsttype) \ - REGISTER_KERNEL_BUILDER(Name("Cast") \ - .TypeConstraint<srctype>("SrcT") \ - .TypeConstraint<dsttype>("DstT") \ + REGISTER_KERNEL_BUILDER(Name("Cast") \ + .TypeConstraint<srctype>("SrcT") \ + .TypeConstraint<dsttype>("DstT") \ .Device(DEVICE_SYCL), \ SyclCastOp) - CURRY_TYPES2(REGISTER_CAST_SYCL, bool); CURRY_TYPES2(REGISTER_CAST_SYCL, int32); CURRY_TYPES2(REGISTER_CAST_SYCL, int64); |