| /external/eigen/unsupported/Eigen/src/SpecialFunctions/arch/CUDA/ |
| D | CudaSpecialFunctions.h | 66 using numext::polygamma;
67 …return make_float4(polygamma(n.x, x.x), polygamma(n.y, x.y), polygamma(n.z, x.z), polygamma(n.w, x…
73 using numext::polygamma;
74 return make_double2(polygamma(n.x, x.x), polygamma(n.y, x.y));
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_Polygamma.pbtxt | 3 summary: "Compute the polygamma function \\\\(\\psi^{(n)}(x)\\\\)."
5 The polygamma function is defined as:
11 The polygamma function is defined only for non-negative integer orders \\a\\.
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_Polygamma.pbtxt | 4 name: "math.polygamma"
7 name: "polygamma"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_op_zeta.cc | 20 REGISTER2(BinaryOp, CPU, "Polygamma", functor::polygamma, float, double);
27 REGISTER2(BinaryOp, GPU, "Polygamma", functor::polygamma, float, double);
|
| D | cwise_op_gpu_zeta.cu.cc | 26 DEFINE_BINARY2(polygamma, float, double);
|
| /external/eigen/unsupported/Eigen/src/SpecialFunctions/ |
| D | SpecialFunctionsPacketMath.h | 31 Packet ppolygamma(const Packet& n, const Packet& x) { using numext::polygamma; return polygamma(n, … in ppolygamma()
|
| D | SpecialFunctionsHalf.h | 24 template<> EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Eigen::half polygamma(const Eigen::half& n, const … in polygamma() function
25 return Eigen::half(Eigen::numext::polygamma(static_cast<float>(n), static_cast<float>(x))); in polygamma()
|
| D | SpecialFunctionsFunctors.h | 171 using numext::polygamma; return polygamma(n, x);
|
| D | SpecialFunctionsArrayAPI.h | 70 polygamma(const Eigen::ArrayBase<DerivedN>& n, const Eigen::ArrayBase<DerivedX>& x) in polygamma() function
|
| D | SpecialFunctionsImpl.h | 1525 EIGEN_DEVICE_FUNC inline EIGEN_MATHFUNC_RETVAL(polygamma, Scalar)
1526 polygamma(const Scalar& n, const Scalar& x) {
1527 return EIGEN_MATHFUNC_IMPL(polygamma, Scalar)::run(n, x);
|
| /external/eigen/unsupported/Eigen/ |
| D | SpecialFunctions | 34 * - polygamma
|
| /external/eigen/unsupported/test/ |
| D | special_functions.cpp | 170 CALL_SUBTEST( res = polygamma(n,x); verify_component_wise(res, ref); ); in array_special_functions()
174 CALL_SUBTEST( res = polygamma(n,x); verify_component_wise(res.head(8), ref.head(8)); ); in array_special_functions()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 209 polygamma = _broadcasting_binary_op(math_ops.polygamma) variable
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | special_math_test.py | 59 return math_ops.polygamma(n, x)
322 expected_values = sps.polygamma(n, x)
335 expected_values = sps.polygamma(n, x)
|
| /external/eigen/Eigen/src/plugins/ |
| D | ArrayCwiseBinaryOps.h | 305 polygamma(const EIGEN_CURRENT_STORAGE_BASE_CLASS<DerivedN> &n) const
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | cwise_ops_binary_test.py | 192 math_ops.polygamma):
195 if tf_func in (math_ops.zeta, math_ops.polygamma):
233 self._compareBoth(n_small, x_pos_small, special.polygamma,
234 math_ops.polygamma)
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | chlo_legalize_to_hlo_broadcasts.mlir | 256 // CHECK: chlo.polygamma %[[LHS]], %[[RHS]]
|
| /external/eigen/doc/ |
| D | CoeffwiseMathFunctionsTable.dox | 485 \link Eigen::polygamma polygamma\endlink(n,x);
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | map_vectorization_test.py | 153 return math_ops.polygamma(
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | math_test.py | 169 return math_ops.polygamma(
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.math.pbtxt | 312 name: "polygamma"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.math.pbtxt | 312 name: "polygamma"
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | chlo_ops.td | 370 def HLOClient_PolygammaOp : HLOClient_Op<"polygamma", [NoSideEffect,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_grad.py | 874 partial_x = math_ops.polygamma(array_ops.constant(1, dtype=x.dtype), x)
1154 partial_x = math_ops.polygamma(n + 1, x)
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorBase.h | 160 polygamma(const OtherDerived& other) const { in polygamma() function
|