Please, help us to better know about our user community by answering the following short survey: https://forms.gle/wpyrxWi18ox9Z5ae9
 
Loading...
Searching...
No Matches
NEON/BesselFunctions.h
1#ifndef EIGEN_NEON_BESSELFUNCTIONS_H
2#define EIGEN_NEON_BESSELFUNCTIONS_H
3
4namespace Eigen {
5namespace internal {
6
7#if EIGEN_HAS_ARM64_FP16_VECTOR_ARITHMETIC
8
9#define NEON_HALF_TO_FLOAT_FUNCTIONS(METHOD) \
10template <> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE \
11Packet8hf METHOD<Packet8hf>(const Packet8hf& x) { \
12 const Packet4f lo = METHOD<Packet4f>(vcvt_f32_f16(vget_low_f16(x))); \
13 const Packet4f hi = METHOD<Packet4f>(vcvt_f32_f16(vget_high_f16(x))); \
14 return vcombine_f16(vcvt_f16_f32(lo), vcvt_f16_f32(hi)); \
15} \
16 \
17template <> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE \
18Packet4hf METHOD<Packet4hf>(const Packet4hf& x) { \
19 return vcvt_f16_f32(METHOD<Packet4f>(vcvt_f32_f16(x))); \
20}
21
22NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_i0)
23NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_i0e)
24NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_i1)
25NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_i1e)
26NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_j0)
27NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_j1)
28NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_k0)
29NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_k0e)
30NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_k1)
31NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_k1e)
32NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_y0)
33NEON_HALF_TO_FLOAT_FUNCTIONS(pbessel_y1)
34
35#undef NEON_HALF_TO_FLOAT_FUNCTIONS
36#endif
37
38BF16_PACKET_FUNCTION(Packet4f, Packet4bf, pbessel_i0)
39BF16_PACKET_FUNCTION(Packet4f, Packet4bf, pbessel_i0e)
40BF16_PACKET_FUNCTION(Packet4f, Packet4bf, pbessel_i1)
41BF16_PACKET_FUNCTION(Packet4f, Packet4bf, pbessel_i1e)
42BF16_PACKET_FUNCTION(Packet4f, Packet4bf, pbessel_j0)
43BF16_PACKET_FUNCTION(Packet4f, Packet4bf, pbessel_j1)
44BF16_PACKET_FUNCTION(Packet4f, Packet4bf, pbessel_k0)
45BF16_PACKET_FUNCTION(Packet4f, Packet4bf, pbessel_k0e)
46BF16_PACKET_FUNCTION(Packet4f, Packet4bf, pbessel_k1)
47BF16_PACKET_FUNCTION(Packet4f, Packet4bf, pbessel_k1e)
48BF16_PACKET_FUNCTION(Packet4f, Packet4bf, pbessel_y0)
49BF16_PACKET_FUNCTION(Packet4f, Packet4bf, pbessel_y1)
50
51} // namespace internal
52} // namespace Eigen
53
54#endif // EIGEN_NEON_BESSELFUNCTIONS_H
Namespace containing all symbols from the Eigen library.