Searched refs:getBernoulli (Results 1 – 6 of 6) sorted by relevance
35 if (getBernoulli(0.5f)) { in broadcastOpConstructor()39 } else if (getBernoulli(0.5f)) { in broadcastOpConstructor()50 if (getBernoulli(0.5f)) { in broadcastOpConstructor()
292 shrinkMask[i] = getBernoulli(0.2f); in stridedSliceConstructor()334 beginMask[i] = (begin[i] == 0 && getBernoulli(0.2f)); in stridedSliceFinalizer()335 endMask[i] = (end[i] == 0 && getBernoulli(0.2f)); in stridedSliceFinalizer()343 if (getBernoulli(0.2f)) { in stridedSliceFinalizer()
117 return getBernoulli(0.5f);
480 squeeze[i] = getBernoulli(0.5f); in squeezeOmittedConstructor()
140 if (getBernoulli(prob)) { in generateGraph()159 if (getBernoulli(prob)) return true; in asConstant()
263 inline bool getBernoulli(double p) { in getBernoulli() function