Searched refs:normMax (Results 1 – 2 of 2) sorted by relevance
113 const T normMax = (maxval - mean) / stddev; in TruncatedNormalKernel() local121 const T diff = normMax - normMin; in TruncatedNormalKernel()126 if (!(input_stddev > T(0) && normMin < normMax && in TruncatedNormalKernel()128 Eigen::numext::isfinite(normMax)))) { in TruncatedNormalKernel()131 (normMax >= T(0.))) || in TruncatedNormalKernel()132 ((normMax > kStdDevsInsideBoundsToUseRandnSampler) && in TruncatedNormalKernel()141 if ((randn[i] >= normMin) && randn[i] <= normMax) { in TruncatedNormalKernel()211 bool accept = (u <= g && z < normMax); in TruncatedNormalKernel()
128 const T normMax = (maxval - mean) / stddev; in operator ()() local137 const T diff = normMax - normMin; in operator ()()140 (normMax >= T(0.))) || in operator ()()141 ((normMax > kStdDevsInsideBoundsToUseRandnSampler) && in operator ()()157 (randn_sample[i] <= normMax)) { in operator ()()174 << "normMin=" << normMin << " normMax=" << normMax in operator ()()250 auto accept = (u <= g && z < normMax); in operator ()()