| /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);
24 REGISTER2(BinaryOp, GPU, "Polygamma", functor::polygamma, float, double);
|
| D | cwise_op_gpu_zeta.cu.cc | 23 DEFINE_BINARY2(polygamma, float, double);
|
| D | cwise_ops.h | 977 struct polygamma : base<T, Eigen::internal::scalar_polygamma_op<T>> {};
|
| /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/tensorflow/tensorflow/contrib/labeled_tensor/ |
| D | __init__.py | 101 polygamma = _core.polygamma variable
|
| /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/eigen/unsupported/Eigen/ |
| D | SpecialFunctions | 34 * - polygamma
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | cwise_ops_binary_test.py | 191 math_ops.polygamma):
194 if tf_func in (math_ops.zeta, math_ops.polygamma):
232 self._compareBoth(n_small, x_pos_small, special.polygamma,
233 math_ops.polygamma)
|
| /external/eigen/Eigen/src/plugins/ |
| D | ArrayCwiseBinaryOps.h | 305 polygamma(const EIGEN_CURRENT_STORAGE_BASE_CLASS<DerivedN> &n) const
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| D | core.py | 1200 polygamma = define_binary_op('polygamma', math_ops.polygamma) variable
|
| D | core_test.py | 846 ('polygamma', None, math_ops.polygamma, core.polygamma),
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | math_test.py | 148 return math_ops.polygamma(
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | map_vectorization_test.py | 143 return math_ops.polygamma(
|
| /external/eigen/doc/ |
| D | CoeffwiseMathFunctionsTable.dox | 485 \link Eigen::polygamma polygamma\endlink(n,x);
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_grad.py | 658 return grad * math_ops.polygamma(array_ops.constant(1, dtype=x.dtype), x)
777 partial_x = math_ops.polygamma(n + 1, x)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.math.pbtxt | 284 name: "polygamma"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.math.pbtxt | 284 name: "polygamma"
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorBase.h | 160 polygamma(const OtherDerived& other) const { in polygamma() function
|