message: Work around crash in nvcc by using the same #include files and order as max_pooling, which seems to not trigger the crash.
diff: @@ -17,11 +17,10 @@ limitations under the License.
  #define EIGEN_USE_GPU
  -#include "tensorflow/core/kernels/eye_functor.h"
  -
  -#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
  +#include "tensorflow/core/framework/register_types.h"
  #include "tensorflow/core/framework/tensor_types.h"
  -#include "tensorflow/core/platform/types.h"
  +#include "tensorflow/core/framework/type_traits.h"
  +#include "tensorflow/core/kernels/eye_functor.h"
  #include "tensorflow/core/util/cuda_kernel_helper.h"
  namespace tensorflow {
  