Lines Matching refs:select
244 self.select = P.Select()
255 z = self.select(need_to_reflect, neg_input, x - 1)
273 …x = self.select(self.lessequal(x, 0.0), self.select(self.equal(abs_frac_input, 0.0), infinity, x),…
274 reduced_frac_input = self.select(self.greater(abs_frac_input, 0.5),
278 reflection = self.select(self.isfinite(reflection_denom),
282 result = self.select(need_to_reflect, reflection, log_y)
284 return self.select(self.isfinite(x), result, infinity)
347 self.select = P.Select()
357 z = self.select(need_to_reflect, neg_input, x - 1)
376 real_result = self.select(need_to_reflect, reflection, y)
379 return self.select(self.logicaland(self.less(x, 0), self.equal(x, self.floor(x))),
400 select = P.Select()
425 return (conditional, select(enabled, r, vals[1]),
426 select(enabled, c, vals[2]), select(enabled, ans, vals[3]),
427 select(enabled, x, vals[4]), select(enabled, dc_da, vals[5]),
428 select(enabled, dans_da, vals[6]))
450 select = P.Select()
488 t = select(qk_is_nonzero, abs_x((ans - r) / r), fill(dtype(t), shape(t), 1))
489 ans = select(qk_is_nonzero, r, ans)
493 dans_da_new = select(qk_is_nonzero, (dpk_da - ans * dqk_da) / qk, dans_da)
494 grad_conditional = select(qk_is_nonzero,
509 pkm2 = select(rescale, pkm2 * epsilon, pkm2)
510 pkm1 = select(rescale, pkm1 * epsilon, pkm1)
511 qkm2 = select(rescale, qkm2 * epsilon, qkm2)
512 qkm1 = select(rescale, qkm1 * epsilon, qkm1)
514 dpkm2_da = select(rescale, dpkm2_da * epsilon, dpkm2_da)
515 dqkm2_da = select(rescale, dqkm2_da * epsilon, dqkm2_da)
516 dpkm1_da = select(rescale, dpkm1_da * epsilon, dpkm1_da)
517 dqkm1_da = select(rescale, dqkm1_da * epsilon, dqkm1_da)
521 return (conditional, select(enabled, ans, vals[1]), select(enabled, t, vals[2]),
522 select(enabled, y, vals[3]), select(enabled, z, vals[4]),
523 c, select(enabled, pkm1, vals[6]),
524 select(enabled, qkm1, vals[7]), select(enabled, pkm2, vals[8]),
525 select(enabled, qkm2, vals[9]), select(enabled, dpkm2_da, vals[10]),
526 select(enabled, dqkm2_da, vals[11]), select(enabled, dpkm1_da, vals[12]),
527 select(enabled, dqkm1_da, vals[13]), select(enabled, dans_da_new, vals[14]))
608 self.select = P.Select()
635 output = self.select(use_igammac,
638 output = self.select(x_is_zero, self.zeroslike(output), output)
639 output = self.select(domain_error, self.fill(self.dtype(a), self.shape(a), np.nan), output)
698 self.select = P.Select()
716 x_min = self.select(comp_less, x, y)
717 y_max = self.select(comp_less, y, x)
748 temp = self.select(comp_yless8, log_beta_small, log_beta_one_large)
749 return self.select(comp_xless8, temp, log_beta_two_large)