math/opencl_2prim_2dot__product_8hpp_source.html

#ifndef STAN_MATH_OPENCL_PRIM_DOT_PRODUCT_HPP

#define STAN_MATH_OPENCL_PRIM_DOT_PRODUCT_HPP

#ifdef STAN_OPENCL

#include <stan/math/opencl/kernel_generator.hpp>

#include <stan/math/opencl/prim/sum.hpp>

#include <stan/math/prim/meta.hpp>

#include <stan/math/prim/err/check_vector.hpp>

#include <stan/math/prim/err/check_matching_sizes.hpp>


namespace stan {

namespace math {


template <typename T_a, typename T_b,

          require_all_kernel_expressions_and_none_scalar_t<T_a, T_b>* = nullptr>

inline auto dot_product(const T_a& a, const T_b& b) {

  const char* function = "dot_product(OpenCL)";

  check_vector(function, "a", a);

  check_vector(function, "b", b);

  return sum(elt_multiply(a, b));

}


}  // namespace math

}  // namespace stan


#endif

#endif

check_matching_sizes.hpp

check_vector.hpp

stan::math::elt_multiply
elt_multiply_< as_operation_cl_t< T_a >, as_operation_cl_t< T_b > > elt_multiply(T_a &&a, T_b &&b)
Definition binary_operation.hpp:204

kernel_generator.hpp

stan::math::check_vector
void check_vector(const char *function, const char *name, const Mat &x)
Check the input is either a row vector or column vector or a matrix with a single row or column.
Definition check_vector.hpp:32

stan::math::sum
auto sum(const std::vector< T > &m)
Return the sum of the entries of the specified standard vector.
Definition sum.hpp:23

stan::math::dot_product
auto dot_product(const T_a &a, const T_b &b)
Returns the dot product of the specified vectors.
Definition dot_product.hpp:26

stan
The lgamma implementation in stan-math is based on either the reentrant safe lgamma_r implementation ...
Definition unit_vector_constrain.hpp:15

sum.hpp

meta.hpp