math/opencl_2prim_2crossprod_8hpp_source.html

#ifndef STAN_MATH_OPENCL_PRIM_FUN_CROSSPROD_HPP

#define STAN_MATH_OPENCL_PRIM_FUN_CROSSPROD_HPP

#ifdef STAN_OPENCL


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

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

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


namespace stan {

namespace math {

template <typename T_A,

          typename = require_all_kernel_expressions_and_none_scalar_t<T_A>>

inline matrix_cl<typename std::decay_t<T_A>::Scalar> crossprod(T_A&& A) {

  const matrix_cl<typename std::decay_t<T_A>::Scalar>& A_eval

      = transpose(std::forward<T_A>(A));

  return multiply_transpose(A_eval);

}

}  // namespace math

}  // namespace stan

#endif

#endif

stan::math::matrix_cl
Represents an arithmetic matrix on the OpenCL device.
Definition matrix_cl.hpp:47

stan::math::transpose
auto transpose(Arg &&a)
Transposes a kernel generator expression.
Definition transpose.hpp:139

stan::math::multiply_transpose
matrix_cl< T > multiply_transpose(const matrix_cl< T > &A)
Computes the product of a square OpenCL matrix with its transpose.
Definition multiply_transpose.hpp:24

kernel_generator.hpp

matrix_cl.hpp

stan::math::crossprod
matrix_cl< typename std::decay_t< T_A >::Scalar > crossprod(T_A &&A)
Returns the result of pre-multiplying a matrix by its own transpose.
Definition crossprod.hpp:21

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

multiply.hpp