LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
lbann::AbsOperator< DataT, D > Class Template Referencefinal

Entrywise absolute value. More...

#include <abs.hpp>

Inheritance diagram for lbann::AbsOperator< DataT, D >:
[legend]
Collaboration diagram for lbann::AbsOperator< DataT, D >:
[legend]

Public Member Functions

Lifecycle
 AbsOperator ()=default
 
 AbsOperator (AbsOperator &&)=default
 
 AbsOperator (AbsOperator const &)=default
 
AbsOperatoroperator= (AbsOperator &&)=default
 
AbsOperatoroperator= (AbsOperator const &)=default
 
 ~AbsOperator ()=default
 
Queries
std::string get_type () const final
 
int get_backprop_requirements () const final
 
Serialization
template<typename ArchiveT >
void serialize (ArchiveT &ar)
 
- Public Member Functions inherited from lbann::Cloneable< AbsOperator< DataT, D >, ElementwiseOperator< DataT, El::Base< DataT >, D > >
std::unique_ptr< AbsOperator< DataT, D > > clone () const
 Return an exception-safe, memory-safe copy of this object. More...
 

Private Types

Private Types
using BaseType = Cloneable< AbsOperator< DataT, D >, ElementwiseOperator< DataT, El::Base< DataT >, D > >
 
using LocalInputTensorType = typename BaseType::LocalInputTensorType
 
using LocalOutputTensorType = typename BaseType::LocalOutputTensorType
 
using ConstLocalInputTensorType = typename BaseType::ConstLocalInputTensorType
 
using ConstLocalOutputTensorType = typename BaseType::ConstLocalOutputTensorType
 

Private Member Functions

virtual void fp_compute_local (std::vector< ConstLocalInputTensorType > input, std::vector< LocalOutputTensorType > output) const final
 Local forward compute function. More...
 
void bp_compute_local (std::vector< ConstLocalInputTensorType > input, std::vector< ConstLocalOutputTensorType > gradient_wrt_output, std::vector< LocalInputTensorType > gradient_wrt_input) const final
 Local backward compute function. More...
 
void set_proto_params (lbann_data::Operator &msg) const final
 
void do_fill_description (description &desc) const final
 

Detailed Description

template<typename DataT, El::Device D>
class lbann::AbsOperator< DataT, D >

Entrywise absolute value.

\[ \text{abs}(x) = |x| \]

Definition at line 51 of file abs.hpp.

Member Typedef Documentation

◆ BaseType

template<typename DataT , El::Device D>
using lbann::AbsOperator< DataT, D >::BaseType = Cloneable<AbsOperator<DataT, D>, ElementwiseOperator<DataT, El::Base<DataT>, D> >
private

Definition at line 59 of file abs.hpp.

◆ ConstLocalInputTensorType

template<typename DataT , El::Device D>
using lbann::AbsOperator< DataT, D >::ConstLocalInputTensorType = typename BaseType::ConstLocalInputTensorType
private

Definition at line 64 of file abs.hpp.

◆ ConstLocalOutputTensorType

template<typename DataT , El::Device D>
using lbann::AbsOperator< DataT, D >::ConstLocalOutputTensorType = typename BaseType::ConstLocalOutputTensorType
private

Definition at line 66 of file abs.hpp.

◆ LocalInputTensorType

template<typename DataT , El::Device D>
using lbann::AbsOperator< DataT, D >::LocalInputTensorType = typename BaseType::LocalInputTensorType
private

Definition at line 61 of file abs.hpp.

◆ LocalOutputTensorType

template<typename DataT , El::Device D>
using lbann::AbsOperator< DataT, D >::LocalOutputTensorType = typename BaseType::LocalOutputTensorType
private

Definition at line 62 of file abs.hpp.

Constructor & Destructor Documentation

◆ AbsOperator() [1/3]

template<typename DataT , El::Device D>
lbann::AbsOperator< DataT, D >::AbsOperator ( )
default

◆ AbsOperator() [2/3]

template<typename DataT , El::Device D>
lbann::AbsOperator< DataT, D >::AbsOperator ( AbsOperator< DataT, D > &&  )
default

◆ AbsOperator() [3/3]

template<typename DataT , El::Device D>
lbann::AbsOperator< DataT, D >::AbsOperator ( AbsOperator< DataT, D > const &  )
default

◆ ~AbsOperator()

template<typename DataT , El::Device D>
lbann::AbsOperator< DataT, D >::~AbsOperator ( )
default

Member Function Documentation

◆ bp_compute_local()

template<typename DataT , El::Device D>
void lbann::AbsOperator< DataT, D >::bp_compute_local ( std::vector< ConstLocalInputTensorType input,
std::vector< ConstLocalOutputTensorType gradient_wrt_output,
std::vector< LocalInputTensorType gradient_wrt_input 
) const
finalprivate

Local backward compute function.

Here is the caller graph for this function:

◆ do_fill_description()

template<typename DataT , El::Device D>
void lbann::AbsOperator< DataT, D >::do_fill_description ( description desc) const
inlinefinalprivate

Definition at line 122 of file abs.hpp.

◆ fp_compute_local()

template<typename DataT , El::Device D>
virtual void lbann::AbsOperator< DataT, D >::fp_compute_local ( std::vector< ConstLocalInputTensorType input,
std::vector< LocalOutputTensorType output 
) const
finalprivatevirtual

Local forward compute function.

Here is the caller graph for this function:

◆ get_backprop_requirements()

template<typename DataT , El::Device D>
int lbann::AbsOperator< DataT, D >::get_backprop_requirements ( ) const
inlinefinal

Definition at line 86 of file abs.hpp.

◆ get_type()

template<typename DataT , El::Device D>
std::string lbann::AbsOperator< DataT, D >::get_type ( ) const
inlinefinal

Definition at line 85 of file abs.hpp.

◆ operator=() [1/2]

template<typename DataT , El::Device D>
AbsOperator& lbann::AbsOperator< DataT, D >::operator= ( AbsOperator< DataT, D > &&  )
default

◆ operator=() [2/2]

template<typename DataT , El::Device D>
AbsOperator& lbann::AbsOperator< DataT, D >::operator= ( AbsOperator< DataT, D > const &  )
default

◆ serialize()

template<typename DataT , El::Device D>
template<typename ArchiveT >
void lbann::AbsOperator< DataT, D >::serialize ( ArchiveT &  ar)
inline

Definition at line 96 of file abs.hpp.

Here is the call graph for this function:

◆ set_proto_params()

template<typename DataT , El::Device D>
void lbann::AbsOperator< DataT, D >::set_proto_params ( lbann_data::Operator &  msg) const
inlinefinalprivate

Definition at line 117 of file abs.hpp.


The documentation for this class was generated from the following file: