entrywise_operator.hpp File Reference

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

Include dependency graph for entrywise_operator.hpp:

Go to the source code of this file.

Namespaces
	lbann

Functions
template<template< typename > class Op, typename TensorDataType >
void	lbann::apply_entrywise_unary_operator (const El::AbstractMatrix< TensorDataType > &input, El::AbstractMatrix< TensorDataType > &output)
template<template< typename > class Op, typename TensorDataType >
void	lbann::apply_entrywise_binary_operator (const El::AbstractMatrix< TensorDataType > &input1, const El::AbstractMatrix< TensorDataType > &input2, El::AbstractMatrix< TensorDataType > &output)
template<template< typename > class Op, typename TensorDataType >
void	lbann::apply_entrywise_unary_operator (const El::AbstractDistMatrix< TensorDataType > &input, El::AbstractDistMatrix< TensorDataType > &output)
template<template< typename > class Op, typename TensorDataType >
void	lbann::apply_entrywise_binary_operator (const El::AbstractDistMatrix< TensorDataType > &input1, const El::AbstractDistMatrix< TensorDataType > &input2, El::AbstractDistMatrix< TensorDataType > &output)