Searched refs:reflection_denom (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | chlo_legalize_to_hlo.cc | 616 Value reflection_denom = rewriter.create<mhlo::LogOp>( in MaterializeLgamma() local
625 reflection_denom), in MaterializeLgamma()
631 rewriter.create<mhlo::IsFiniteOp>(loc, reflection_denom); in MaterializeLgamma()
633 rewriter.create<mhlo::NegOp>(loc, reflection_denom); in MaterializeLgamma()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 692 Value reflection_denom = in matchAndRewrite() local
698 rewriter.create<IsFiniteOp>(loc, tensor_bool_type, reflection_denom); in matchAndRewrite()
699 Value neg_reflection_denom = rewriter.create<NegOp>(loc, reflection_denom); in matchAndRewrite()
701 rewriter.create<SubOp>(loc, log_pi, reflection_denom); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | math.cc | 591 XlaOp reflection_denom = Log(Sin(pi * reduced_frac_input)); in Lgamma() local
596 Select(IsFinite(reflection_denom), log_pi - reflection_denom - log_y, in Lgamma()
597 -reflection_denom); in Lgamma()
|