diff options
Diffstat (limited to 'third_party/eigen3/unsupported/Eigen/CXX11/NeuralNetworks')
| -rw-r--r-- | third_party/eigen3/unsupported/Eigen/CXX11/NeuralNetworks | 35 |
1 files changed, 35 insertions, 0 deletions
diff --git a/third_party/eigen3/unsupported/Eigen/CXX11/NeuralNetworks b/third_party/eigen3/unsupported/Eigen/CXX11/NeuralNetworks new file mode 100644 index 0000000000..7741b68d8a --- /dev/null +++ b/third_party/eigen3/unsupported/Eigen/CXX11/NeuralNetworks @@ -0,0 +1,35 @@ +// This file is part of Eigen, a lightweight C++ template library +// for linear algebra. +// +// Copyright (C) 2014 Benoit Steiner <benoit.steiner.goog@gmail.com> +// +// This Source Code Form is subject to the terms of the Mozilla +// Public License v. 2.0. If a copy of the MPL was not distributed +// with this file, You can obtain one at http://mozilla.org/MPL/2.0/. + +#ifndef EIGEN_CXX11_NEURAL_NETWORKS_MODULE +#define EIGEN_CXX11_NEURAL_NETWORKS_MODULE + +#include "unsupported/Eigen/CXX11/Tensor" + +/** \defgroup CXX11_NeuralNetworks_Module Neural Networks Module + * + * This module provides an efficient implementation of the common primitives + * used by neural networks. + * The primitives are built on top of the tensor library. + * + * \code + * #include <Eigen/CXX11/NeuralNetworks> + * \endcode + */ + +#include "unsupported/Eigen/CXX11/src/NeuralNetworks/Activations.h" +#include "unsupported/Eigen/CXX11/src/NeuralNetworks/Attention.h" +#include "unsupported/Eigen/CXX11/src/NeuralNetworks/Pooling.h" +#include "unsupported/Eigen/CXX11/src/NeuralNetworks/SoftMax.h" +#include "unsupported/Eigen/CXX11/src/NeuralNetworks/BackwardCuboidConvolutions.h" +#include "unsupported/Eigen/CXX11/src/NeuralNetworks/CuboidConvolution.h" +#include "unsupported/Eigen/CXX11/src/NeuralNetworks/BackwardSpatialConvolutions.h" +#include "unsupported/Eigen/CXX11/src/NeuralNetworks/SpatialConvolutions.h" + +#endif // EIGEN_CXX11_NEURAL_NETWORKS_MODULE |