diff options
Diffstat (limited to 'tensorflow/core/kernels/tile_ops.cc')
| -rw-r--r-- | tensorflow/core/kernels/tile_ops.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/tile_ops.cc b/tensorflow/core/kernels/tile_ops.cc index 06f20cd9ec..7c72487d3f 100644 --- a/tensorflow/core/kernels/tile_ops.cc +++ b/tensorflow/core/kernels/tile_ops.cc @@ -265,7 +265,9 @@ TF_CALL_complex128(HANDLE_TYPE_NAME_GPU); #ifdef TENSORFLOW_USE_SYCL TF_CALL_float(HANDLE_TYPE_NAME_SYCL); TF_CALL_double(HANDLE_TYPE_NAME_SYCL); +TF_CALL_int16(HANDLE_TYPE_NAME_SYCL); TF_CALL_int32(HANDLE_TYPE_NAME_SYCL); +TF_CALL_int64(HANDLE_TYPE_NAME_SYCL); #endif // TENSORFLOW_USE_SYCL #undef HANDLE_TYPE_NAME_CPU @@ -522,7 +524,9 @@ TF_CALL_complex128(HANDLE_TYPE_NAME_GPU); TF_CALL_float(HANDLE_TYPE_NAME_SYCL); TF_CALL_double(HANDLE_TYPE_NAME_SYCL); +TF_CALL_int16(HANDLE_TYPE_NAME_SYCL); TF_CALL_int32(HANDLE_TYPE_NAME_SYCL); +TF_CALL_int64(HANDLE_TYPE_NAME_SYCL); #undef HANDLE_TYPE_NAME_SYCL #endif // TENSORFLOW_USE_SYCL |