Searched refs:normMin (Results 1 – 2 of 2) sorted by relevance
112 const T normMin = (minval - mean) / stddev; in TruncatedNormalKernel() local116 const T sqrtFactor = Eigen::numext::sqrt((normMin * normMin) + T(4)); in TruncatedNormalKernel()119 Eigen::numext::exp(T(0.5) + (normMin * (normMin - sqrtFactor)) / T(4)) / in TruncatedNormalKernel()120 (normMin + sqrtFactor); in TruncatedNormalKernel()121 const T diff = normMax - normMin; in TruncatedNormalKernel()126 if (!(input_stddev > T(0) && normMin < normMax && in TruncatedNormalKernel()127 (Eigen::numext::isfinite(normMin) || in TruncatedNormalKernel()130 } else if (((normMin < -kStdDevsInsideBoundsToUseRandnSampler) && in TruncatedNormalKernel()133 (normMin <= T(0.)))) { in TruncatedNormalKernel()141 if ((randn[i] >= normMin) && randn[i] <= normMax) { in TruncatedNormalKernel()[all …]
127 const T normMin = (minval - mean) / stddev; in operator ()() local131 const T sqrtFactor = Eigen::numext::sqrt((normMin * normMin) + T(4)); in operator ()()135 (normMin * (normMin - sqrtFactor)) / T(4)) / in operator ()()136 (normMin + sqrtFactor); in operator ()()137 const T diff = normMax - normMin; in operator ()()139 if (((normMin < -kStdDevsInsideBoundsToUseRandnSampler) && in operator ()()142 (normMin <= T(0.)))) { in operator ()()156 if ((randn_sample[i] >= normMin) && in operator ()()174 << "normMin=" << normMin << " normMax=" << normMax in operator ()()187 const T plusFactor = (normMin < T(0)) ? T(0) : normMin * normMin; in operator ()()[all …]