| /external/tensorflow/tensorflow/core/kernels/ |
| cwise_op_gpu_digamma.cu.cc | 22 DEFINE_UNARY3(digamma, Eigen::half, float, double);
|
| cwise_op_digamma.cc | 19 REGISTER3(UnaryOp, CPU, "Digamma", functor::digamma, float, Eigen::half,
22 REGISTER3(UnaryOp, GPU, "Digamma", functor::digamma, float, Eigen::half,
|
| cwise_ops.h | 611 struct digamma : base<T, Eigen::internal::scalar_digamma_op<T>> {}; struct in namespace:tensorflow::functor
[all...] |
| /external/eigen/unsupported/Eigen/src/SpecialFunctions/ |
| SpecialFunctionsPacketMath.h | 23 Packet pdigamma(const Packet& a) { using numext::digamma; return digamma(a); }
|
| SpecialFunctionsHalf.h | 18 template<> EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Eigen::half digamma(const Eigen::half& a) { function in namespace:Eigen::numext
19 return Eigen::half(Eigen::numext::digamma(static_cast<float>(a)));
|
| SpecialFunctionsFunctors.h | 122 * \sa class CwiseUnaryOp, Cwise::digamma()
127 using numext::digamma; return digamma(a);
|
| SpecialFunctionsImpl.h | 148 * Implementation of digamma (psi), based on Cephes *
158 * Polynomial evaluation helper for the Psi (digamma) function.
228 * Psi (digamma) function (modified for Eigen)
1027 // Just return the digamma function for n = 1
1514 digamma(const Scalar& x) { function in namespace:Eigen::numext
[all...] |
| /external/eigen/unsupported/Eigen/src/SpecialFunctions/arch/CUDA/ |
| CudaSpecialFunctions.h | 38 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/ |
| beta.py | 255 - (self.concentration1 - 1.) * math_ops.digamma(self.concentration1)
256 - (self.concentration0 - 1.) * math_ops.digamma(self.concentration0)
258 math_ops.digamma(self.total_concentration)))
364 - math_ops.digamma(d1.concentration1) * delta("concentration1")
365 - math_ops.digamma(d1.concentration0) * delta("concentration0")
366 + (math_ops.digamma(d1.total_concentration)
|
| dirichlet.py | 223 * math_ops.digamma(self.total_concentration))
225 (self.concentration - 1.) * math_ops.digamma(self.concentration),
|
| gamma.py | 221 math_ops.digamma(self.concentration)))
302 * math_ops.digamma(g0.concentration))
|
| student_t.py | 270 (math_ops.digamma(0.5 * (self.df + 1.)) -
271 math_ops.digamma(0.5 * self.df)))
|
| /external/apache-commons-math/src/main/java/org/apache/commons/math/special/ |
| Gamma.java | 64 // limits for switching algorithm in digamma
275 * <p>Computes the digamma function of x.</p>
278 * Jose Bernardo, Algorithm AS 103: Psi (Digamma) Function, Applied Statistics, 1976.</p>
289 * @return digamma(x) to within 10-8 relative or absolute error whichever is smaller
290 * @see <a href="http://en.wikipedia.org/wiki/Digamma_function"> Digamma at wikipedia </a>
294 public static double digamma(double x) { method in class:Gamma
310 return digamma(x + 1) - 1 / x;
315 * the implementation of digamma.</p>
320 * @see Gamma#digamma(double)
|
| /external/eigen/unsupported/test/ |
| special_functions.cpp | 45 VERIFY_IS_APPROX(m1.digamma(), digamma(m1));
147 // digamma
154 CALL_SUBTEST( res = x.digamma(); verify_component_wise(res, ref); );
155 CALL_SUBTEST( res = digamma(x); verify_component_wise(res, ref); );
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| kumaraswamy.py | 59 return math_ops.digamma(x + one) - math_ops.digamma(one)
|
| inverse_gamma.py | 220 math_ops.digamma(self.concentration)))
|
| wishart.py | 432 """Computes the multivariate digamma function; Psi_p(a)."""
435 return math_ops.reduce_sum(math_ops.digamma(seq),
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/ |
| __init__.py | 71 digamma = _core.digamma variable
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| beta_test.py | 379 (a1 - a2) * special.digamma(a1) +
380 (b1 - b2) * special.digamma(b1) +
381 (a2 - a1 + b2 - b1) * special.digamma(a1 + b1))
|
| gamma_test.py | 394 kl_expected = ((alpha0 - alpha1) * special.digamma(alpha0)
|
| /external/eigen/Eigen/src/Core/ |
| GlobalFunctions.h | 70 EIGEN_ARRAY_DECLARE_GLOBAL_UNARY(digamma,scalar_digamma_op,derivative of lgamma,\sa ArrayBase::digamma)
|
| /external/eigen/Eigen/src/plugins/ |
| ArrayCwiseUnaryOps.h | 486 * \sa <a href="group__CoeffwiseMathFunctions.html#cwisetable_lgamma">Math functions</a>, digamma()
495 /** \returns an expression of the coefficient-wise digamma (psi, derivative of lgamma).
500 * the user has to provide implementations of digamma(T) for any scalar
503 * \sa <a href="group__CoeffwiseMathFunctions.html#cwisetable_digamma">Math functions</a>, Eigen::digamma(), Eigen::polygamma(), lgamma()
507 digamma() const function
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| core.py | 1120 digamma = define_unary_op('digamma', math_ops.digamma) variable
[all...] |
| core_test.py | 730 ('digamma', None, math_ops.digamma, core.digamma),
|
| /external/tensorflow/tensorflow/contrib/bayesflow/python/kernel_tests/ |
| hmc_test.py | 222 expected_x = (math_ops.digamma(self._shape_param)
[all...] |
