operators/math/common.hpp File Reference

#include "lbann/base.hpp"
#include "lbann/utils/profiling.hpp"

Include dependency graph for operators/math/common.hpp:

Go to the source code of this file.

Namespaces
	lbann
	lbann::internal

Functions
template<typename S , typename T , typename U , typename F >
void	lbann::internal::EntrywiseZipInto (El::Matrix< S, El::Device::CPU > const &A, El::Matrix< T, El::Device::CPU > const &B, El::Matrix< U, El::Device::CPU > &C, F func)
	A binary entrywise map c <- f(a,b). More...
template<typename DataT , typename F >
void	lbann::internal::apply_binary_backprop_operator (El::Matrix< DataT, El::Device::CPU > const &x1, El::Matrix< DataT, El::Device::CPU > const &x2, El::Matrix< DataT, El::Device::CPU > const &dy, El::Matrix< DataT, El::Device::CPU > &dx1, El::Matrix< DataT, El::Device::CPU > &dx2, F f)
template<typename S , typename T , typename U , typename R , typename F >
void	lbann::internal::EntrywiseZipInto (El::Matrix< S, El::Device::CPU > const &A, El::Matrix< T, El::Device::CPU > const &B, El::Matrix< U, El::Device::CPU > const &C, El::Matrix< R, El::Device::CPU > &D, F func)
	A ternary entrywise map c <- f(a,b,c). More...
template<typename DataT , typename F >
void	lbann::internal::apply_ternary_backprop_operator (El::Matrix< DataT, El::Device::CPU > const &x1, El::Matrix< DataT, El::Device::CPU > const &x2, El::Matrix< DataT, El::Device::CPU > const &x3, El::Matrix< DataT, El::Device::CPU > const &dy, El::Matrix< DataT, El::Device::CPU > &dx1, El::Matrix< DataT, El::Device::CPU > &dx2, El::Matrix< DataT, El::Device::CPU > &dx3, F f)