From 7617682a46b14840c6c05957970380976e1b8564 Mon Sep 17 00:00:00 2001 From: shugeo Date: Wed, 16 Oct 2019 14:45:10 +0300 Subject: [PATCH] Added declarations for igamma and igammac ops. --- .../ops/declarable/headers/broadcastable.h | 22 +++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/libnd4j/include/ops/declarable/headers/broadcastable.h b/libnd4j/include/ops/declarable/headers/broadcastable.h index b3b2463cd..7ee53b52a 100644 --- a/libnd4j/include/ops/declarable/headers/broadcastable.h +++ b/libnd4j/include/ops/declarable/headers/broadcastable.h @@ -357,6 +357,28 @@ namespace nd4j { #if NOT_EXCLUDED(OP_Pow) DECLARE_BROADCASTABLE_OP(Pow, 0, 0); #endif + + /** + * Broadcastable igamma implementation + * + * igamma(a, x) = gamma(а, x) / Gamma(a) - Gamma distribution function P(a,x) + * Gamma(a) = int from 0 to infinity { t ^ {a - 1} e^{-t}dt } + * gamma(a, x) = int from 0 to x { t ^ {a - 1} e^{-t}dt } + * @tparam T + */ + #if NOT_EXCLUDED(OP_igamma) + DECLARE_BROADCASTABLE_OP(igamma, 0, 0); + #endif + /** + * Broadcastable igammac implementation + * igammac(a, x) = Gamma(a,x)/Gamma(а) - Gamma distribution function Q(a,x) + * Gamma(a) = int from 0 to infinity { t ^ {a - 1} e^{-t}dt } + * Gamma(a, x) = int from x to infinity { t ^ {a - 1} e^{-t}dt } + * @tparam T + */ + #if NOT_EXCLUDED(OP_igammac) + DECLARE_BROADCASTABLE_OP(igammac, 0, 0); + #endif } }