Fixed incorrect debug assertion

Change: 127903711

1 files changed, 1 insertions, 1 deletions

diff --git a/tensorflow/core/kernels/eigen_spatial_convolutions.h b/tensorflow/core/kernels/eigen_spatial_convolutions.h
index 8925e9a58f..a2a9e15301 100644
--- a/tensorflow/core/kernels/eigen_spatial_convolutions.h
+++ b/tensorflow/core/kernels/eigen_spatial_convolutions.h
@@ -715,7 +715,7 @@ struct gemm_pack_rhs<
                   std::min<Index>(peeled_k - c * patch_rows * patch_depth -
                                       r * patch_depth + startDepth,
                                   patch_depth);
-              eigen_assert(max_depth % packet_size == 0);
+              eigen_assert((max_depth - startDepth) % packet_size == 0);
               for (Index d = startDepth; d < max_depth; d += packet_size) {
                 eigen_assert(k < peeled_k);
                 PacketBlock<Packet, 4> kernel;