lukeiwanski / tensorflow

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

[OpenCL] Provides atomic free MaxPool gradients #110

Closed jwlawson closed 7 years ago

jwlawson commented 7 years ago

Atomic support in SYCL is not designed in a way that plays nicely with Tensorflow and Eigen. Here we provide a new implementation for MaxPoolGrad and MaxPool3DGrad which does not rely on atomics, and so avoids any such problems.