/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/tensorexpr
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/tensorexpr/cpp_intrinsics.h (719B)
#pragma once
namespace torch {
namespace jit {
namespace tensorexpr {
constexpr auto cpp_intrinsics_definition = R"(
namespace std {
template
::value, int>::type = 0>
T rsqrt(T v) {
return 1.0f / std::sqrt(v);
}
template ::value, int>::type = 0>
T frac(T v) {
T intpart;
return std::modf(v, &intpart);
}
template
To bitcast(const From& v) {
assert(sizeof(To) == sizeof(From));
To res;
std::memcpy(&res, &v, sizeof(From));
return res;
}
} // namespace std
)";
} // namespace tensorexpr
} // namespace jit
} // namespace torch