Searched refs:iteration_bcast (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | math.cc | 1766 auto iteration_bcast = Broadcast(iteration, shape.dimensions()); in RegularizedIncompleteBeta() local
1768 Eq(iteration_bcast % FullLike(iteration_bcast, 2), in RegularizedIncompleteBeta()
1769 FullLike(iteration_bcast, 0)); in RegularizedIncompleteBeta()
1770 auto iteration_is_one = Eq(iteration_bcast, FullLike(iteration_bcast, 1)); in RegularizedIncompleteBeta()
1771 auto iteration_minus_one = iteration_bcast - FullLike(iteration_bcast, 1); in RegularizedIncompleteBeta()
1791 auto iteration_bcast = Broadcast(iteration, shape.dimensions()); in RegularizedIncompleteBeta() local
1793 Select(Eq(iteration_bcast, ScalarLike(iteration_bcast, 0)), in RegularizedIncompleteBeta()
|