From aea0766bf913f9149aa601294663ed0519887183 Mon Sep 17 00:00:00 2001 From: Esun Kim Date: Thu, 21 May 2026 11:05:32 -0700 Subject: [PATCH] Removed batch_size assert in depthwise_conv --- tensorflow/lite/micro/kernels/cmsis_nn/depthwise_conv.cc | 1 - 1 file changed, 1 deletion(-) diff --git a/tensorflow/lite/micro/kernels/cmsis_nn/depthwise_conv.cc b/tensorflow/lite/micro/kernels/cmsis_nn/depthwise_conv.cc index 7183a28a70b..0db1cf04388 100644 --- a/tensorflow/lite/micro/kernels/cmsis_nn/depthwise_conv.cc +++ b/tensorflow/lite/micro/kernels/cmsis_nn/depthwise_conv.cc @@ -147,7 +147,6 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { const int batch_size = MatchingDim(input_shape, 0, output_shape, 0); const int output_depth = MatchingDim(output_shape, 3, filter_shape, 3); - TFLITE_DCHECK_EQ(batch_size, 1); /* Only batch = 1 is supported */ cmsis_nn_dims input_dims; input_dims.n = batch_size;