diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-02-21 11:50:59 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-21 11:55:21 -0800 |
commit | fec6ce3aee1e6a65b1dbd2c4364b59179050b703 (patch) | |
tree | 711d478d98d2603cc22e477db6f68c1551937219 /tensorflow/contrib/lite/kernels/conv.cc | |
parent | 47042733daef84843e8e573920ebdeebb4ef04ef (diff) |
Adding support for dilated convolution.
PiperOrigin-RevId: 186496353
Diffstat (limited to 'tensorflow/contrib/lite/kernels/conv.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/conv.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/kernels/conv.cc b/tensorflow/contrib/lite/kernels/conv.cc index b2fdd61dc0..b93a416351 100644 --- a/tensorflow/contrib/lite/kernels/conv.cc +++ b/tensorflow/contrib/lite/kernels/conv.cc @@ -371,7 +371,7 @@ void EvalFloat(TfLiteContext* context, TfLiteNode* node, reference_ops::Conv(GetTensorData<float>(input), GetTensorDims(input), GetTensorData<float>(filter), GetTensorDims(filter), GetTensorData<float>(bias), GetTensorDims(bias), - params->stride_width, params->stride_height, + params->stride_width, params->stride_height, 1, 1, data->padding.width, data->padding.height, output_activation_min, output_activation_max, GetTensorData<float>(output), GetTensorDims(output), @@ -382,7 +382,7 @@ void EvalFloat(TfLiteContext* context, TfLiteNode* node, optimized_ops::Conv(GetTensorData<float>(input), GetTensorDims(input), GetTensorData<float>(filter), GetTensorDims(filter), GetTensorData<float>(bias), GetTensorDims(bias), - params->stride_width, params->stride_height, + params->stride_width, params->stride_height, 1, 1, data->padding.width, data->padding.height, output_activation_min, output_activation_max, GetTensorData<float>(output), GetTensorDims(output), |