Searched refs:use_igammac (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | math.cc | 932 XlaOp use_igammac = And(Gt(x, ScalarLike(x, 1)), Gt(x, a)); in Igamma() local
939 use_igammac, in Igamma()
941 ax, x, a, And(enabled, use_igammac), type), in Igamma()
942 IgammaSeries<VALUE>(ax, x, a, And(enabled, Not(use_igammac)), type)); in Igamma()
979 XlaOp use_igammac = And(Gt(x, ScalarLike(x, 1)), Gt(x, a)); in IgammaGradA() local
985 XlaOp output = Select(use_igammac, in IgammaGradA()
987 ax, x, a, And(enabled, use_igammac), type), in IgammaGradA()
989 ax, x, a, And(enabled, Not(use_igammac)), type)); in IgammaGradA()
1026 XlaOp use_igammac = And(Gt(x, ScalarLike(x, 1)), Gt(x, a)); in RandomGammaGrad() local
1032 XlaOp output = Select(use_igammac, in RandomGammaGrad()
[all …]
|