From 309bb709a6f36c8e2c9cd48123fdcfb7a754e7aa Mon Sep 17 00:00:00 2001 From: aarushgupta Date: Sat, 3 Mar 2018 22:38:30 +0530 Subject: [PATCH] Some minor changes --- .../ann/convolution_rules/naive_atrous_convolution.hpp | 4 ++-- src/mlpack/methods/ann/layer/atrous_convolution.hpp | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/mlpack/methods/ann/convolution_rules/naive_atrous_convolution.hpp b/src/mlpack/methods/ann/convolution_rules/naive_atrous_convolution.hpp index e92464e62c..622c030feb 100644 --- a/src/mlpack/methods/ann/convolution_rules/naive_atrous_convolution.hpp +++ b/src/mlpack/methods/ann/convolution_rules/naive_atrous_convolution.hpp @@ -111,8 +111,8 @@ class NaiveAtrousConvolution inputPadded.submat((filter.n_rows - 1)*dilation + filter.n_rows - 1, (filter.n_cols - 1)*dilation + filter.n_cols - 1, (filter.n_rows - 1)*dilation + filter.n_rows - 1 + input.n_rows - 1, - (filter.n_cols - 1)*dilation + filter.n_cols - 1 + input.n_cols - 1 - ) = input; + (filter.n_cols - 1)*dilation + filter.n_cols - 1 + input.n_cols - 1) + = input; NaiveAtrousConvolution::Convolution(inputPadded, filter, output, 1, 1, dilation); diff --git a/src/mlpack/methods/ann/layer/atrous_convolution.hpp b/src/mlpack/methods/ann/layer/atrous_convolution.hpp index 69bb3d9ffd..0848a23e7d 100644 --- a/src/mlpack/methods/ann/layer/atrous_convolution.hpp +++ b/src/mlpack/methods/ann/layer/atrous_convolution.hpp @@ -186,7 +186,7 @@ class AtrousConvolution * @param d The dilation size. * @return The convolution output size. */ - size_t ConvOutSize( const size_t size, + size_t ConvOutSize (const size_t size, const size_t k, const size_t s, const size_t p,