Searched refs:tie_low (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | arithmetic_test.cc | 37 bool tie_low) { in TestArgMin() argument
38 TestArgMinMax(input, expected_output, axis, /*is_min=*/true, tie_low); in TestArgMin()
44 bool tie_low) { in TestArgMax() argument
45 TestArgMinMax(input, expected_output, axis, /*is_min=*/false, tie_low); in TestArgMax()
54 bool tie_low) { in TestArgMinMax() argument
58 return ArgMin(op, type, axis, /*stable=*/true, tie_low); in TestArgMinMax()
62 return ArgMinTwoPass(op, type, axis, tie_low); in TestArgMinMax()
67 return ArgMax(op, type, axis, /*stable=*/true, tie_low); in TestArgMinMax()
71 return ArgMaxTwoPass(op, type, axis, tie_low); in TestArgMinMax()
|
| D | arithmetic.cc | 118 bool stable, bool tie_low) { in CreateMinMaxComputation() argument
136 tie_low ? Min(lhs_index, rhs_index) : Max(lhs_index, rhs_index); in CreateMinMaxComputation()
144 bool stable, bool tie_low) { in ArgMinMax() argument
163 is_min, stable, tie_low); in ArgMinMax()
176 bool is_min, bool tie_low) { in ArgMinMaxTwoPass() argument
197 if (tie_low) { in ArgMinMaxTwoPass()
221 bool tie_low) { in ArgMax() argument
222 return ArgMinMax(input, output_type, axis, /*is_min=*/false, stable, tie_low); in ArgMax()
226 bool tie_low) { in ArgMin() argument
227 return ArgMinMax(input, output_type, axis, /*is_min=*/true, stable, tie_low); in ArgMin()
[all …]
|
| D | arithmetic.h | 85 bool stable = false, bool tie_low = true);
87 bool tie_low = true);
95 bool stable = false, bool tie_low = true);
97 bool tie_low = true);
|