Skip to content

Commit

Permalink
[OpenCL] Fix for //tensorflow/python/kernel_tests:image_ops_test (#111)…
Browse files Browse the repository at this point in the history
… (#12041)

* [OpenCL] Fix for //tensorflow/python/kernel_tests:image_ops_test

* Conforming to Google style comments

* [OpenCL] Fixed RGBToHSVTest
  • Loading branch information
Luke Iwanski authored and rmlarsen committed Aug 7, 2017
1 parent 4ff7190 commit af5e87d
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions tensorflow/core/kernels/unpack_op.cc
Expand Up @@ -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,
Expand All @@ -85,6 +87,7 @@ class UnpackOp : public OpKernel {
}
return;
}
#endif // TENSORFLOW_USE_SYCL

int64 before_dim = 1;
for (int i = 0; i < axis; ++i) {
Expand Down

0 comments on commit af5e87d

Please sign in to comment.