diff --git a/tensorflow/core/kernels/conv_2d_gpu.h b/tensorflow/core/kernels/conv_2d_gpu.h index 90d85e6f04e..297016160ad 100644 --- a/tensorflow/core/kernels/conv_2d_gpu.h +++ b/tensorflow/core/kernels/conv_2d_gpu.h @@ -210,7 +210,7 @@ __global__ void ShuffleInTensor3Simple(int nthreads, } } -constexpr int kUnroll = 4; +static constexpr int kUnroll = 4; template __global__ void ShuffleInTensor3SimpleVector(int nthreads, @@ -246,7 +246,7 @@ __global__ void ShuffleInTensor3SimpleVector(int nthreads, *out = *reinterpret_cast(buf); } - for(; output_index < nthreads; output_index++) { + for (; output_index < nthreads; ++output_index) { Index<3> output_tensor_index = FlatToTensorIndex(output_index, output_dims); Index<3> input_tensor_index;