diff --git a/src/gallium/frontends/teflon/tfl_device.c b/src/gallium/frontends/teflon/tfl_device.c
index aa08f09b861..bde022b90d7 100644
--- a/src/gallium/frontends/teflon/tfl_device.c
+++ b/src/gallium/frontends/teflon/tfl_device.c
@@ -109,14 +109,24 @@ fill_operation(struct teflon_delegate *delegate, TfLiteContext *tf_context, TfLi
    switch(node_registration->builtin_code) {
       case kTfLiteBuiltinConv2d:
       case kTfLiteBuiltinDepthwiseConv2d: {
-         TfLiteConvParams* params = (TfLiteConvParams*)node->builtin_data;
          operation->type = PIPE_ML_OPERATION_TYPE_CONVOLUTION;
          operation->conv.weight_tensor = &tensors[node->inputs->data[1]];
          operation->conv.bias_tensor = &tensors[node->inputs->data[2]];
-         operation->conv.stride_x = params->stride_width;
-         operation->conv.stride_y = params->stride_height;
-         operation->conv.padding_same = params->padding == kTfLitePaddingSame;
-         operation->conv.depthwise = node_registration->builtin_code == kTfLiteBuiltinDepthwiseConv2d;
+         if (node_registration->builtin_code == kTfLiteBuiltinConv2d) {
+            TfLiteConvParams* params = (TfLiteConvParams*)node->builtin_data;
+
+            operation->conv.stride_x = params->stride_width;
+            operation->conv.stride_y = params->stride_height;
+            operation->conv.padding_same = params->padding == kTfLitePaddingSame;
+            operation->conv.depthwise = false;
+         } else {
+            TfLiteDepthwiseConvParams* params = (TfLiteDepthwiseConvParams*)node->builtin_data;
+
+            operation->conv.stride_x = params->stride_width;
+            operation->conv.stride_y = params->stride_height;
+            operation->conv.padding_same = params->padding == kTfLitePaddingSame;
+            operation->conv.depthwise = true;
+         }
          operation->conv.pointwise = operation->conv.weight_tensor->dims[1] == 1 && \
                                      operation->conv.weight_tensor->dims[2] == 1;
          break;