diff options
Diffstat (limited to 'tensorflow/core/kernels/unpack_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/unpack_op.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/unpack_op.cc b/tensorflow/core/kernels/unpack_op.cc index c3bebfcbf9..7fd1def1fe 100644 --- a/tensorflow/core/kernels/unpack_op.cc +++ b/tensorflow/core/kernels/unpack_op.cc @@ -69,6 +69,8 @@ class UnpackOp : public OpKernel { std::numeric_limits<Eigen::DenseIndex>::max()), errors::InvalidArgument("output size must fit in Eigen DenseIndex")); +// This optimization is currently not applicable for SYCL devices +#ifndef TENSORFLOW_USE_SYCL // Special case: Aligned, so we can share the underlying buffer. // // Apply this optimization conservatively: if input is aligned, @@ -85,6 +87,7 @@ class UnpackOp : public OpKernel { } return; } +#endif // TENSORFLOW_USE_SYCL int64 before_dim = 1; for (int i = 0; i < axis; ++i) { |