lukeiwanski / tensorflow

OpenCL support for TensorFlow via SYCL
Apache License 2.0
65 stars 14 forks source link

[OpenCL] SYCL gather functor needs indices on host #169

Closed jwlawson closed 7 years ago

jwlawson commented 7 years ago

Changes the registration of the ResourceGatherOp to ensure that the tensor storing the indices is kept in host memory, as this is required by the current SYCL implementation of GatherFunctor.