| /external/tensorflow/tensorflow/core/kernels/ |
| cwise_op_gpu_zeta.cu.cc | 23 DEFINE_BINARY2(polygamma, float, double);
|
| cwise_op_zeta.cc | 20 REGISTER2(BinaryOp, CPU, "Polygamma", functor::polygamma, float, double);
24 REGISTER2(BinaryOp, GPU, "Polygamma", functor::polygamma, float, double);
|
| cwise_ops.h | 977 struct polygamma : base<T, Eigen::internal::scalar_polygamma_op<T>> {}; struct in namespace:tensorflow::functor
[all...] |
| /external/eigen/unsupported/Eigen/src/SpecialFunctions/ |
| SpecialFunctionsPacketMath.h | 29 /** \internal \returns the polygamma function (coeff-wise) */
31 Packet ppolygamma(const Packet& n, const Packet& x) { using numext::polygamma; return polygamma(n, x); }
|
| SpecialFunctionsHalf.h | 24 template<> EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Eigen::half polygamma(const Eigen::half& n, const Eigen::half& x) { function in namespace:Eigen::numext
25 return Eigen::half(Eigen::numext::polygamma(static_cast<float>(n), static_cast<float>(x)));
|
| SpecialFunctionsArrayAPI.h | 56 /** \cpp11 \returns an expression of the coefficient-wise polygamma(\a n, \a x) to the given arrays.
61 * or float/double in non c++11 mode, the user has to provide implementations of polygamma(T,T) for any scalar
66 // * \warning Be careful with the order of the parameters: x.polygamma(n) is equivalent to polygamma(n,x)
67 // * \sa ArrayBase::polygamma()
70 polygamma(const Eigen::ArrayBase<DerivedN>& n, const Eigen::ArrayBase<DerivedX>& x) function in namespace:Eigen
|
| SpecialFunctionsFunctors.h | 165 * \brief Template functor to compute the polygamma function.
166 * \sa class CwiseUnaryOp, Cwise::polygamma()
171 using numext::polygamma; return polygamma(n, x);
|
| SpecialFunctionsImpl.h | 958 * the polygamma function.
993 * Implementation of polygamma function, requires C++11/C99 *
1526 polygamma(const Scalar& n, const Scalar& x) { function in namespace:Eigen::numext
[all...] |
| /external/eigen/unsupported/Eigen/src/SpecialFunctions/arch/CUDA/ |
| 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.w));
73 using numext::polygamma;
74 return make_double2(polygamma(n.x, x.x), polygamma(n.y, x.y));
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/ |
| __init__.py | 101 polygamma = _core.polygamma variable
|
| /external/eigen/unsupported/test/ |
| special_functions.cpp | 169 // CALL_SUBTEST( res = x.polygamma(n); verify_component_wise(res, ref); );
170 CALL_SUBTEST( res = polygamma(n,x); verify_component_wise(res, ref); );
173 // CALL_SUBTEST( res = x.polygamma(n); verify_component_wise(res.head(8), ref.head(8)); );
174 CALL_SUBTEST( res = polygamma(n,x); verify_component_wise(res.head(8), ref.head(8)); );
|
| /external/eigen/Eigen/src/plugins/ |
| ArrayCwiseBinaryOps.h | 293 /** \cpp11 \returns an expression of the coefficient-wise polygamma function.
299 * \warning Be careful with the order of the parameters: x.polygamma(n) is equivalent to polygamma(n,x)
301 * \sa Eigen::polygamma()
305 polygamma(const EIGEN_CURRENT_STORAGE_BASE_CLASS<DerivedN> &n) const
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| 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)
[all...] |
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| core.py | 1200 polygamma = define_binary_op('polygamma', math_ops.polygamma) variable
[all...] |
| core_test.py | 846 ('polygamma', None, math_ops.polygamma, core.polygamma),
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| map_vectorization_test.py | 141 # Wrapper functions restricting the range of inputs of zeta and polygamma.
143 return math_ops.polygamma(
170 ("Polygamma", safe_polygamma),
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| math_test.py | 146 # Wrapper functions restricting the range of inputs of zeta and polygamma.
148 return math_ops.polygamma(
|
| pfor.py | [all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| math_grad.py | 658 return grad * math_ops.polygamma(array_ops.constant(1, dtype=x.dtype), x)
763 @ops.RegisterGradient("Polygamma")
777 partial_x = math_ops.polygamma(n + 1, x)
[all...] |
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorBase.h | 157 // polygamma(n = this, x = other)
160 polygamma(const OtherDerived& other) const { function in class:Eigen::TensorBase
[all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
