Searched refs:elem_operand (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_typed_visitor.h | 195 ElementWiseUnaryOp(abs, [](NativeT elem_operand) {
196 return elem_operand;
205 ElementWiseUnaryOp(abs, [](NativeT elem_operand) {
206 return std::abs(elem_operand);
220 abs, [](NativeT elem_operand) { return std::abs(elem_operand); },
243 ElementWiseUnaryOp(round, [](ElementwiseT elem_operand) {
244 return std::round(elem_operand);
267 ElementWiseUnaryOp(round, [](ElementwiseT elem_operand) {
268 return std::nearbyint(elem_operand);
289 ElementWiseUnaryOp(ceil, [](ElementwiseT elem_operand) {
[all …]
|
| D | hlo_evaluator.cc | 1213 [](Eigen::half elem_operand) { in HandleIsFinite() argument
1214 return std::isfinite(static_cast<float>(elem_operand)); in HandleIsFinite()
1223 [](bfloat16 elem_operand) { in HandleIsFinite() argument
1224 return std::isfinite(static_cast<float>(elem_operand)); in HandleIsFinite()
1233 [](float elem_operand) { return std::isfinite(elem_operand); }, in HandleIsFinite() argument
1241 [](double elem_operand) { return std::isfinite(elem_operand); }, in HandleIsFinite() argument
1256 real, [](bfloat16 elem_operand) { return elem_operand; }, in HandleReal() argument
1263 real, [](complex64 elem_operand) { return std::real(elem_operand); }, in HandleReal() argument
1270 real, [](complex128 elem_operand) { return std::real(elem_operand); }, in HandleReal() argument
1277 real, [](Eigen::half elem_operand) { return elem_operand; }, in HandleReal() argument
[all …]
|