| /external/eigen/unsupported/Eigen/src/SpecialFunctions/arch/CUDA/ |
| D | 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/core/api_def/python_api/ |
| D | api_def_Digamma.pbtxt | 4 name: "math.digamma"
7 name: "digamma"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_op_digamma.cc | 20 REGISTER3(UnaryOp, CPU, "Digamma", functor::digamma, float, Eigen::half,
25 REGISTER3(UnaryOp, GPU, "Digamma", functor::digamma, float, Eigen::half,
|
| D | cwise_op_gpu_digamma.cu.cc | 23 DEFINE_UNARY3(digamma, Eigen::half, float, double);
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | beta.py | 294 - (self.concentration1 - 1.) * math_ops.digamma(self.concentration1)
295 - (self.concentration0 - 1.) * math_ops.digamma(self.concentration0)
297 math_ops.digamma(self.total_concentration)))
408 - math_ops.digamma(d1.concentration1) * delta("concentration1")
409 - math_ops.digamma(d1.concentration0) * delta("concentration0")
410 + (math_ops.digamma(d1.total_concentration)
|
| D | dirichlet.py | 262 * math_ops.digamma(self.total_concentration))
264 (self.concentration - 1.) * math_ops.digamma(self.concentration),
405 digamma_sum_d1 = math_ops.digamma(
407 digamma_diff = math_ops.digamma(d1.concentration) - digamma_sum_d1
|
| D | gamma.py | 251 math_ops.digamma(self.concentration)))
337 * math_ops.digamma(g0.concentration))
|
| D | student_t.py | 298 (math_ops.digamma(0.5 * (self.df + 1.)) -
299 math_ops.digamma(0.5 * self.df)))
|
| /external/eigen/unsupported/Eigen/src/SpecialFunctions/ |
| D | SpecialFunctionsPacketMath.h | 23 Packet pdigamma(const Packet& a) { using numext::digamma; return digamma(a); } in pdigamma()
|
| D | SpecialFunctionsHalf.h | 18 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()
|
| D | SpecialFunctionsFunctors.h | 127 using numext::digamma; return digamma(a);
|
| D | SpecialFunctionsImpl.h | 1513 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/ |
| D | special_functions.cpp | 45 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/ |
| D | Gamma.java | 294 public static double digamma(double x) { in digamma() method in Gamma
310 return digamma(x + 1) - 1 / x; in digamma()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_Polygamma.pbtxt | 10 where \\(\psi(x)\\) is the digamma function.
|
| /external/eigen/unsupported/Eigen/ |
| D | SpecialFunctions | 33 * - digamma
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_test_ops.py | 44 math_ops.digamma,
|
| /external/eigen/Eigen/src/Core/ |
| D | GlobalFunctions.h | 70 …EIGEN_ARRAY_DECLARE_GLOBAL_UNARY(digamma,scalar_digamma_op,derivative of lgamma,\sa ArrayBase::dig…
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| D | beta_test.py | 382 (a1 - a2) * special.digamma(a1) +
383 (b1 - b2) * special.digamma(b1) +
384 (a2 - a1 + b2 - b1) * special.digamma(a1 + b1))
|
| D | dirichlet_test.py | 299 + np.sum((conc1 - conc2) * (special.digamma(conc1) - special.digamma(
|
| D | gamma_test.py | 393 kl_expected = ((alpha0 - alpha1) * special.digamma(alpha0)
|
| /external/eigen/doc/ |
| D | CoeffwiseMathFunctionsTable.dox | 445 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/compiler/tf2xla/python/ |
| D | xla.py | 84 digamma = _unary_op(math_ops.digamma) variable
|
| /external/eigen/Eigen/src/plugins/ |
| D | ArrayCwiseUnaryOps.h | 517 digamma() const in digamma() function
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | chlo_legalize_to_hlo.cc | 715 Value digamma = rewriter.create<mhlo::SubOp>( in MaterializeDigamma() local
740 loc, digamma, in MaterializeDigamma()
745 digamma = rewriter.create<mhlo::SelectOp>(loc, need_to_reflect, reflection, in MaterializeDigamma()
746 digamma); in MaterializeDigamma()
761 digamma); in MaterializeDigamma()
|