Searched refs:elem_operand (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_typed_visitor.h | 157 ElementWiseUnaryOp(abs, [](NativeT elem_operand) { in HandleAbs()
158 return elem_operand; in HandleAbs()
168 ElementWiseUnaryOp(abs, [](NativeT elem_operand) { in HandleAbs()
169 return std::abs(elem_operand); in HandleAbs()
183 abs, [](NativeT elem_operand) { return std::abs(elem_operand); }, in HandleAbs()
207 ElementWiseUnaryOp(round, [](ElementwiseT elem_operand) { in HandleRound()
208 return std::round(elem_operand); in HandleRound()
229 ElementWiseUnaryOp(ceil, [](ElementwiseT elem_operand) { in HandleCeil()
230 return std::ceil(elem_operand); in HandleCeil()
269 ElementWiseUnaryOp(exp, [](ElementwiseT elem_operand) { in HandleExp()
[all …]
|
| D | hlo_evaluator.cc | 522 [](Eigen::half elem_operand) { in HandleIsFinite() argument
523 return std::isfinite(static_cast<float>(elem_operand)); in HandleIsFinite()
532 [](bfloat16 elem_operand) { in HandleIsFinite() argument
533 return std::isfinite(static_cast<float>(elem_operand)); in HandleIsFinite()
542 [](float elem_operand) { return std::isfinite(elem_operand); }, in HandleIsFinite() argument
550 [](double elem_operand) { return std::isfinite(elem_operand); }, in HandleIsFinite() argument
565 real, [](bfloat16 elem_operand) { return elem_operand; }, in HandleReal() argument
572 real, [](complex64 elem_operand) { return std::real(elem_operand); }, in HandleReal() argument
579 real, [](complex128 elem_operand) { return std::real(elem_operand); }, in HandleReal() argument
586 real, [](Eigen::half elem_operand) { return elem_operand; }, in HandleReal() argument
[all …]
|