mlpack
set_input_width_visitor.hpp
Go to the documentation of this file.
1 
13 #ifndef MLPACK_METHODS_ANN_VISITOR_SET_INPUT_WIDTH_VISITOR_HPP
14 #define MLPACK_METHODS_ANN_VISITOR_SET_INPUT_WIDTH_VISITOR_HPP
15 
16 #include <mlpack/methods/ann/layer/layer_traits.hpp>
17 
18 #include <boost/variant.hpp>
19 
20 namespace mlpack {
21 namespace ann {
22 
27 class SetInputWidthVisitor : public boost::static_visitor<bool>
28 {
29  public:
31  SetInputWidthVisitor(const size_t inputWidth = 0, const bool reset = false);
32 
34  template<typename LayerType>
35  bool operator()(LayerType* layer) const;
36 
37  bool operator()(MoreTypes layer) const;
38 
39  private:
41  size_t inputWidth;
42 
44  bool reset;
45 
48  template<typename T>
49  typename std::enable_if<
50  !HasInputWidth<T, size_t&(T::*)()>::value &&
51  !HasModelCheck<T>::value, bool>::type
52  LayerInputWidth(T* layer) const;
53 
55  template<typename T>
56  typename std::enable_if<
57  HasInputWidth<T, size_t&(T::*)()>::value &&
58  !HasModelCheck<T>::value, bool>::type
59  LayerInputWidth(T* layer) const;
60 
62  template<typename T>
63  typename std::enable_if<
64  !HasInputWidth<T, size_t&(T::*)()>::value &&
65  HasModelCheck<T>::value, bool>::type
66  LayerInputWidth(T* layer) const;
67 
70  template<typename T>
71  typename std::enable_if<
72  HasInputWidth<T, size_t&(T::*)()>::value &&
73  HasModelCheck<T>::value, bool>::type
74  LayerInputWidth(T* layer) const;
75 };
76 
77 } // namespace ann
78 } // namespace mlpack
79 
80 // Include implementation.
81 #include "set_input_width_visitor_impl.hpp"
82 
83 #endif
mlpack::ann::SetInputWidthVisitor::SetInputWidthVisitor
SetInputWidthVisitor(const size_t inputWidth=0, const bool reset=false)
Update the input width parameter with the given input width.
mlpack::ann::MoreTypes
boost::variant< Recurrent< arma::mat, arma::mat > *, RecurrentAttention< arma::mat, arma::mat > *, ReinforceNormal< arma::mat, arma::mat > *, Reparametrization< arma::mat, arma::mat > *, Select< arma::mat, arma::mat > *, Sequential< arma::mat, arma::mat, false > *, Sequential< arma::mat, arma::mat, true > *, Subview< arma::mat, arma::mat > *, VRClassReward< arma::mat, arma::mat > *, VirtualBatchNorm< arma::mat, arma::mat > *> MoreTypes
Definition: layer_types.hpp:189
layer_traits.hpp
mlpack
strip_type.hpp
Definition: add_to_po.hpp:21
mlpack::ann::SetInputWidthVisitor
SetInputWidthVisitor updates the input width parameter with the given input width.
Definition: set_input_width_visitor.hpp:27
mlpack::ann::SetInputWidthVisitor::operator()
bool operator()(LayerType *layer) const
Update the input width parameter.