Searched refs:need_to_reflect (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | math.cc | 331 XlaOp need_to_reflect = Lt(input, one_half); in Lgamma() local
332 XlaOp z = Select(need_to_reflect, -input, input - one); in Lgamma()
401 XlaOp result = Select(need_to_reflect, reflection, log_y); in Lgamma()
442 XlaOp need_to_reflect = Lt(input, one_half); in Digamma() local
443 XlaOp z = Select(need_to_reflect, -input, input - one); in Digamma()
474 XlaOp real_result = Select(need_to_reflect, reflection, y); in Digamma()
|