Home
last modified time | relevance | path

Searched defs:polygamma (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/eigen/unsupported/Eigen/src/SpecialFunctions/
H A DSpecialFunctionsArrayAPI.h112 polygamma(const Eigen::ArrayBase<DerivedN>& n, const Eigen::ArrayBase<DerivedX>& x) in polygamma() function
H A DSpecialFunctionsHalf.h24 template<> EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Eigen::half polygamma(const Eigen::half& n, const … in polygamma() function
H A DSpecialFunctionsBFloat16.h24 template<> EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Eigen::bfloat16 polygamma(const Eigen::bfloat16& n… in polygamma() function
/aosp_15_r20/external/pytorch/torch/special/
H A D__init__.py153 polygamma = _add_docstr(_special.special_polygamma, variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/python/
H A Dxla.py209 polygamma = _broadcasting_binary_op(math_ops.polygamma) variable
/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorBase.h248 polygamma(const OtherDerived& other) const { in polygamma() function
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/
H A Dspecial.h89 inline Tensor polygamma(int64_t n, const Tensor& self) { in polygamma() function
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dops_handler.py614 def polygamma(self, x: T, y: T) -> T: member in OpsHandler
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dgeneric_layout_optimizer_transposer_test.cc2701 auto polygamma = ops::Polygamma( in TEST_F() local