Searched refs:reduced_input (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | arithmetic.cc | 192 XlaOp reduced_input = Reduce(input, init_value, reducer, in ArgMinMaxTwoPass() local
199 XlaOp select_mask = Select(Eq(input, reduced_input, broadcast_dims), in ArgMinMaxTwoPass()
208 XlaOp select_mask = Select(Eq(input, reduced_input, broadcast_dims), in ArgMinMaxTwoPass()
|
| D | math.cc | 692 XlaOp reduced_input = input + Abs(Floor(input + ScalarLike(input, 0.5))); in Digamma() local
694 y - pi * Cos(pi * reduced_input) / Sin(pi * reduced_input); in Digamma()
|