Home
last modified time | relevance | path

Searched refs:digamma (Results 1 – 25 of 46) sorted by relevance

12

/external/eigen/unsupported/Eigen/src/SpecialFunctions/arch/CUDA/
DCudaSpecialFunctions.h38 using numext::digamma;
39 return make_float4(digamma(a.x), digamma(a.y), digamma(a.z), digamma(a.w));
45 using numext::digamma;
46 return make_double2(digamma(a.x), digamma(a.y));
/external/tensorflow/tensorflow/python/ops/distributions/
Dbeta.py290 - (self.concentration1 - 1.) * math_ops.digamma(self.concentration1)
291 - (self.concentration0 - 1.) * math_ops.digamma(self.concentration0)
293 math_ops.digamma(self.total_concentration)))
404 - math_ops.digamma(d1.concentration1) * delta("concentration1")
405 - math_ops.digamma(d1.concentration0) * delta("concentration0")
406 + (math_ops.digamma(d1.total_concentration)
Ddirichlet.py258 * math_ops.digamma(self.total_concentration))
260 (self.concentration - 1.) * math_ops.digamma(self.concentration),
402 digamma_sum_d1 = math_ops.digamma(
404 digamma_diff = math_ops.digamma(d1.concentration) - digamma_sum_d1
Dgamma.py248 math_ops.digamma(self.concentration)))
334 * math_ops.digamma(g0.concentration))
Dstudent_t.py295 (math_ops.digamma(0.5 * (self.df + 1.)) -
296 math_ops.digamma(0.5 * self.df)))
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_Digamma.pbtxt4 name: "math.digamma"
7 name: "digamma"
/external/tensorflow/tensorflow/core/kernels/
Dcwise_op_digamma.cc19 REGISTER3(UnaryOp, CPU, "Digamma", functor::digamma, float, Eigen::half,
22 REGISTER3(UnaryOp, GPU, "Digamma", functor::digamma, float, Eigen::half,
Dcwise_op_gpu_digamma.cu.cc22 DEFINE_UNARY3(digamma, Eigen::half, float, double);
/external/eigen/unsupported/Eigen/src/SpecialFunctions/
DSpecialFunctionsPacketMath.h23 Packet pdigamma(const Packet& a) { using numext::digamma; return digamma(a); } in pdigamma()
DSpecialFunctionsHalf.h18 template<> EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Eigen::half digamma(const Eigen::half& a) { in digamma() function
19 return Eigen::half(Eigen::numext::digamma(static_cast<float>(a))); in digamma()
DSpecialFunctionsFunctors.h127 using numext::digamma; return digamma(a);
DSpecialFunctionsImpl.h1513 EIGEN_DEVICE_FUNC inline EIGEN_MATHFUNC_RETVAL(digamma, Scalar)
1514 digamma(const Scalar& x) {
1515 return EIGEN_MATHFUNC_IMPL(digamma, Scalar)::run(x);
/external/eigen/unsupported/test/
Dspecial_functions.cpp45 VERIFY_IS_APPROX(m1.digamma(), digamma(m1)); in array_special_functions()
154 CALL_SUBTEST( res = x.digamma(); verify_component_wise(res, ref); ); in array_special_functions()
155 CALL_SUBTEST( res = digamma(x); verify_component_wise(res, ref); ); in array_special_functions()
/external/apache-commons-math/src/main/java/org/apache/commons/math/special/
DGamma.java294 public static double digamma(double x) { in digamma() method in Gamma
310 return digamma(x + 1) - 1 / x; in digamma()
/external/tensorflow/tensorflow/contrib/distributions/python/ops/
Dkumaraswamy.py67 return math_ops.digamma(x + one) - math_ops.digamma(one)
Dinverse_gamma.py225 math_ops.digamma(self.concentration)))
/external/tensorflow/tensorflow/contrib/labeled_tensor/
D__init__.py71 digamma = _core.digamma variable
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_Polygamma.pbtxt10 where \\(\psi(x)\\) is the digamma function.
/external/eigen/unsupported/Eigen/
DSpecialFunctions33 * - digamma
/external/eigen/Eigen/src/Core/
DGlobalFunctions.h70 …EIGEN_ARRAY_DECLARE_GLOBAL_UNARY(digamma,scalar_digamma_op,derivative of lgamma,\sa ArrayBase::dig…
/external/eigen/doc/
DCoeffwiseMathFunctionsTable.dox445 a.\link ArrayBase::digamma digamma\endlink(); \n
446 \link Eigen::digamma digamma\endlink(a);
487 …<td><a href="https://en.wikipedia.org/wiki/Polygamma_function">n-th derivative of digamma at x</a>…
490 <a href="#cwisetable_digamma"> \c digamma </a>
/external/tensorflow/tensorflow/python/kernel_tests/distributions/
Dbeta_test.py370 (a1 - a2) * special.digamma(a1) +
371 (b1 - b2) * special.digamma(b1) +
372 (a2 - a1 + b2 - b1) * special.digamma(a1 + b1))
Ddirichlet_test.py297 + np.sum((conc1 - conc2) * (special.digamma(conc1) - special.digamma(
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py81 digamma = _unary_op(math_ops.digamma) variable
/external/eigen/Eigen/src/plugins/
DArrayCwiseUnaryOps.h507 digamma() const in digamma() function

12