math/opencl_2prim_2squared__distance_8hpp_source.html

#ifndef STAN_MATH_OPENCL_PRIM_SQUARED_DISTANCE_HPP

#define STAN_MATH_OPENCL_PRIM_SQUARED_DISTANCE_HPP

#ifdef STAN_OPENCL


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

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

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

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

#include <stan/math/prim/fun/Eigen.hpp>

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

#include <algorithm>


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 squared_distance(const T_a& a, const T_b& b) {

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

  check_vector(function, "a", a);

  check_vector(function, "b", b);

  return sum(square(a - b));

}


}  // namespace math

}  // namespace stan

#endif

#endif

Eigen.hpp

kernel_generator.hpp

matrix_cl.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
fvar< T > sum(const std::vector< fvar< T > > &m)
Return the sum of the entries of the specified standard vector.
Definition sum.hpp:22

stan::math::squared_distance
auto squared_distance(const T_a &a, const T_b &b)
Returns the squared distance.
Definition squared_distance.hpp:30

stan::math::square
fvar< T > square(const fvar< T > &x)
Definition square.hpp:12

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

err.hpp

scalar_type.hpp

meta.hpp