convolution_test.cpp File Reference

#include <mlpack/core.hpp>
#include <mlpack/methods/ann/convolution_rules/border_modes.hpp>
#include <mlpack/methods/ann/convolution_rules/naive_convolution.hpp>
#include <mlpack/methods/ann/convolution_rules/fft_convolution.hpp>
#include <mlpack/methods/ann/convolution_rules/svd_convolution.hpp>
#include "serialization.hpp"
#include "catch.hpp"
#include "test_catch_tools.hpp"

Include dependency graph for convolution_test.cpp:

Functions
template<class ConvolutionFunction >
void	Convolution2DMethodTest (const arma::mat input, const arma::mat filter, const arma::mat output)
template<class ConvolutionFunction >
void	Convolution3DMethodTest (const arma::cube input, const arma::cube filter, const arma::cube output)
template<class ConvolutionFunction >
void	ConvolutionMethodBatchTest (const arma::mat input, const arma::cube filter, const arma::cube output)
	TEST_CASE ("ValidConvolution2DTest", "[ConvolutionTest]")
	Test the convolution (valid) methods.
	TEST_CASE ("FullConvolution2DTest", "[ConvolutionTest]")
	Test the convolution (full) methods.
	TEST_CASE ("ValidConvolution3DTest", "[ConvolutionTest]")
	Test the convolution (valid) methods using 3rd order tensors.
	TEST_CASE ("FullConvolution3DTest", "[ConvolutionTest]")
	Test the convolution (full) methods using 3rd order tensors.
	TEST_CASE ("ValidConvolutionBatchTest", "[ConvolutionTest]")
	Test the convolution (valid) methods using dense matrix as input and a 3rd order tensors as filter and output (batch modus).
	TEST_CASE ("FullConvolutionBatchTest", "[ConvolutionTest]")
	Test the convolution (full) methods using dense matrix as input and a 3rd order tensors as filter and output (batch modus).

Detailed Description

Author

Shangtong Zhang

Marcus Edel

Tests for various convolution strategies.

mlpack is free software; you may redistribute it and/or modify it under the terms of the 3-clause BSD license. You should have received a copy of the 3-clause BSD license along with mlpack. If not, see http://www.opensource.org/licenses/BSD-3-Clause for more information.