Searched refs:need_to_reflect (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | chlo_legalize_to_hlo.cc | 524 Value need_to_reflect = rewriter.create<mhlo::CompareOp>(loc, x, half, kLT); in MaterializeLgamma() local
529 rewriter.create<mhlo::SelectOp>(loc, need_to_reflect, neg_x, x_sub_one); in MaterializeLgamma()
638 lgamma = rewriter.create<mhlo::SelectOp>(loc, need_to_reflect, in MaterializeLgamma()
669 Value need_to_reflect = rewriter.create<mhlo::CompareOp>(loc, x, half, kLT); in MaterializeDigamma() local
674 rewriter.create<mhlo::SelectOp>(loc, need_to_reflect, neg_x, x_sub_one); in MaterializeDigamma()
745 digamma = rewriter.create<mhlo::SelectOp>(loc, need_to_reflect, reflection, in MaterializeDigamma()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | math.cc | 528 XlaOp need_to_reflect = Lt(input, one_half); in Lgamma() local
529 XlaOp z = Select(need_to_reflect, -input, input - one); in Lgamma()
598 XlaOp result = Select(need_to_reflect, reflection, log_y); in Lgamma()
663 XlaOp need_to_reflect = Lt(input, one_half); in Digamma() local
664 XlaOp z = Select(need_to_reflect, -input, input - one); in Digamma()
695 XlaOp real_result = Select(need_to_reflect, reflection, y); in Digamma()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 596 Value need_to_reflect = rewriter.create<LessOp>(loc, input, one_half); in matchAndRewrite() local
597 Type tensor_bool_type = need_to_reflect.getType(); in matchAndRewrite()
598 Value z = rewriter.create<SelectV2Op>(loc, need_to_reflect, minus_input, in matchAndRewrite()
708 rewriter.create<SelectV2Op>(loc, need_to_reflect, reflection, log_y); in matchAndRewrite()
|