Pass in dilation for Depthwise Convolution.
Change-Id: Ib3d83fbe5f3ec62b632e7686c8a2eade8d2df4fb
diff --git a/tensorflow/lite/micro/kernels/kelvin/depthwise_conv.cc b/tensorflow/lite/micro/kernels/kelvin/depthwise_conv.cc
index 5dc2b6d..489fa66 100644
--- a/tensorflow/lite/micro/kernels/kelvin/depthwise_conv.cc
+++ b/tensorflow/lite/micro/kernels/kelvin/depthwise_conv.cc
@@ -121,7 +121,6 @@
if (dw_params.padding_type == PaddingType::kValid &&
dw_params.stride_width == 1 && dw_params.stride_height == 1 &&
filter_shape.Dims(1) == 1 && filter_shape.Dims(2) == 3 &&
- dw_params.dilation_width_factor == 1 &&
dw_params.dilation_height_factor == 1) {
kelvin::opt::DepthwiseConv2DKelvinS16K3x1(
tflite::micro::GetTensorData<int16_t>(input),
@@ -129,6 +128,7 @@
tflite::micro::GetOptionalTensorData<int64_t>(bias),
filter_shape.Dims(3) /*channels*/,
input_shape.Dims(2) /*frames*/,
+ dw_params.dilation_width_factor /*dilation*/,
data.per_channel_output_multiplier,
data.per_channel_output_shift,
dw_params.quantized_activation_min,
