diff options
author | 2018-04-13 08:12:42 -0700 | |
---|---|---|
committer | 2018-04-13 08:16:46 -0700 | |
commit | 91c31997e6854a3d07acc76381cff7436df1c1dd (patch) | |
tree | ee4f38d890771de2f2144c718a6421673eb81caf /tensorflow/contrib/lite/kernels/depthwise_conv.cc | |
parent | f9de043501e401af73aa02ab950864534f07c1df (diff) |
Add support to TFLite for dilated convolution.
PiperOrigin-RevId: 192770919
Diffstat (limited to 'tensorflow/contrib/lite/kernels/depthwise_conv.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/depthwise_conv.cc | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/contrib/lite/kernels/depthwise_conv.cc b/tensorflow/contrib/lite/kernels/depthwise_conv.cc index cad9ce114c..eeda1bc3c5 100644 --- a/tensorflow/contrib/lite/kernels/depthwise_conv.cc +++ b/tensorflow/contrib/lite/kernels/depthwise_conv.cc @@ -140,10 +140,10 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { int out_height = compute_out_size(height, filter_height, params->stride_height); - data->padding.height = - ComputePadding(params->stride_height, height, filter_height, out_height); + data->padding.height = ComputePadding(params->stride_height, 1, height, + filter_height, out_height); data->padding.width = - ComputePadding(params->stride_width, width, filter_width, out_width); + ComputePadding(params->stride_width, 1, width, filter_width, out_width); // Note that quantized inference requires that all tensors have their // parameters set. This is usually done during quantized training. |