Searched refs:elem_operand (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_typed_visitor.h | 190 ElementWiseUnaryOp(abs, [](NativeT elem_operand) {
191 return elem_operand;
201 ElementWiseUnaryOp(abs, [](NativeT elem_operand) {
202 return std::abs(elem_operand);
217 abs, [](NativeT elem_operand) { return std::abs(elem_operand); },
241 ElementWiseUnaryOp(round, [](ElementwiseT elem_operand) {
242 return std::round(elem_operand);
263 ElementWiseUnaryOp(ceil, [](ElementwiseT elem_operand) {
264 return std::ceil(elem_operand);
303 ElementWiseUnaryOp(exp, [](ElementwiseT elem_operand) {
[all …]
|
| D | hlo_evaluator.cc | 559 [](Eigen::half elem_operand) { in HandleIsFinite() argument
560 return std::isfinite(static_cast<float>(elem_operand)); in HandleIsFinite()
569 [](bfloat16 elem_operand) { in HandleIsFinite() argument
570 return std::isfinite(static_cast<float>(elem_operand)); in HandleIsFinite()
579 [](float elem_operand) { return std::isfinite(elem_operand); }, in HandleIsFinite() argument
587 [](double elem_operand) { return std::isfinite(elem_operand); }, in HandleIsFinite() argument
602 real, [](bfloat16 elem_operand) { return elem_operand; }, in HandleReal() argument
609 real, [](complex64 elem_operand) { return std::real(elem_operand); }, in HandleReal() argument
616 real, [](complex128 elem_operand) { return std::real(elem_operand); }, in HandleReal() argument
623 real, [](Eigen::half elem_operand) { return elem_operand; }, in HandleReal() argument
[all …]
|