ann_visitor_test.cpp File Reference

Tests for testing visitors in ANN's of mlpack. More...

#include <mlpack/core.hpp>
#include <mlpack/methods/ann/layer/layer.hpp>
#include <mlpack/methods/ann/layer/layer_types.hpp>
#include <mlpack/methods/ann/visitor/bias_set_visitor.hpp>
#include <mlpack/methods/ann/visitor/weight_set_visitor.hpp>
#include <mlpack/methods/ann/visitor/reset_visitor.hpp>
#include "catch.hpp"
#include "test_catch_tools.hpp"

Include dependency graph for ann_visitor_test.cpp:

Functions
	TEST_CASE ("BiasSetVisitorTest", "[ANNVisitorTest]")
	Test that the BiasSetVisitor works properly.
void	CheckCorrectnessOfWeightSize (LayerTypes<> &layer)
	Check correctness of WeightSize() for a layer.
	TEST_CASE ("WeightSetVisitorTest", "[ANNVisitorTest]")
	Test that WeightSetVisitor works properly.
	TEST_CASE ("WeightSizeVisitorTestForLinearLayer", "[ANNVisitorTest]")
	Test that WeightSizeVisitor works properly for linear layer.
	TEST_CASE ("WeightSizeVisitorTestForConcatLayer", "[ANNVisitorTest]")
	Test that WeightSizeVisitor works properly for concat layer.
	TEST_CASE ("WeightSizeVisitorTestForFastLSTMLayer", "[ANNVisitorTest]")
	Test that WeightSizeVisitor works properly for fast lstm layer.
	TEST_CASE ("WeightSizeVisitorTestForAddLayer", "[ANNVisitorTest]")
	Test that WeightSizeVisitor works properly for Add layer.
	TEST_CASE ("WeightSizeVisitorTestForAtrousConvolutionLayer", "[ANNVisitorTest]")
	Test that WeightSizeVisitor works properly for Atrous Convolution Layer.
	TEST_CASE ("WeightSizeVisitorTestForConvLayer", "[ANNVisitorTest]")
	Test that WeightSizeVisitor works properly for Convolution layer.
	TEST_CASE ("WeightSizeVisitorTestForBatchNormLayer", "[ANNVisitorTest]")
	Test that WeightSizeVisitor works properly for BatchNorm layer.
	TEST_CASE ("WeightSizeVisitorTestForLSTMLayer", "[ANNVisitorTest]")
	Test that WeightSizeVisitor works properly for LSTM layer.
	TEST_CASE ("WeightSizeVisitorTestForTransposedConvLayer", "[ANNVisitorTest]")
	Test that WeightSizeVisitor works properly for Transposed Convolution layer.
	TEST_CASE ("WeightSizeVisitorTestForNoisyLinearLayer", "[ANNVisitorTest]")
	Test that WeightSizeVisitor works properly for noisy linear layer.
	TEST_CASE ("WeightSizeVisitorTestForMultiheadAttentionLayer", "[ANNVisitorTest]")
	Test that WeightSizeVisitor works properly for Multihead Attention layer.

Detailed Description

Tests for testing visitors in ANN's of mlpack.

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.