colwise_reduction.hpp File Reference

#include <stan/math/prim/meta.hpp>
#include <stan/math/opencl/opencl_context.hpp>
#include <stan/math/opencl/matrix_cl_view.hpp>
#include <stan/math/opencl/kernel_generator/type_str.hpp>
#include <stan/math/opencl/kernel_generator/name_generator.hpp>
#include <stan/math/opencl/kernel_generator/operation_cl.hpp>
#include <stan/math/opencl/kernel_generator/as_operation_cl.hpp>
#include <stan/math/opencl/kernel_generator/rowwise_reduction.hpp>
#include <stan/math/opencl/kernel_generator/calc_if.hpp>
#include <map>
#include <string>
#include <type_traits>
#include <utility>

Go to the source code of this file.

Classes
class	stan::math::internal::colwise_reduction_base
class	stan::math::colwise_reduction< Derived, T, Operation >
	Represents a column wise reduction in kernel generator expressions. More...
class	stan::math::colwise_sum_< T >
	Represents column wise sum - reduction in kernel generator expressions. More...
class	stan::math::colwise_prod_< T >
	Represents column wise product - reduction in kernel generator expressions. More...
class	stan::math::colwise_max_< T >
	Represents column wise max - reduction in kernel generator expressions. More...
class	stan::math::colwise_min_< T >
	Represents column wise min - reduction in kernel generator expressions. More...
struct	stan::math::internal::is_colwise_reduction_impl< T >
struct	stan::math::internal::is_colwise_reduction_impl< calc_if_< true, T > >

Namespaces
namespace	stan
	The lgamma implementation in stan-math is based on either the reentrant safe lgamma_r implementation from C or the boost::math::lgamma implementation.
namespace	stan::math
	Matrices and templated mathematical functions.
namespace	stan::math::internal
	A comparator that works for any container type that has the brackets operator.

Typedefs
template<typename T >
using	stan::math::is_colwise_reduction = internal::is_colwise_reduction_impl< std::decay_t< T > >
	Check whether a kernel generator expression is a colwise reduction.

Functions
int	stan::math::internal::colwise_reduction_wgs_rows (int n_rows, int n_cols)
	Determine number of work groups in rows direction that will be run fro colwise reduction of given size.
template<typename T , require_all_kernel_expressions_t< T > * = nullptr>
auto	stan::math::colwise_sum (T &&a)
	Column wise sum - reduction of a kernel generator expression.
template<typename T , require_all_kernel_expressions_t< T > * = nullptr>
auto	stan::math::colwise_prod (T &&a)
	Column wise product - reduction of a kernel generator expression.
template<typename T , require_all_kernel_expressions_t< T > * = nullptr>
auto	stan::math::colwise_max (T &&a)
	Column wise max - reduction of a kernel generator expression.
template<typename T , require_all_kernel_expressions_t< T > * = nullptr>
auto	stan::math::colwise_min (T &&a)
	Column wise min - reduction of a kernel generator expression.