Lines Matching full:complex
3 …"c10/util/complex_math.h is not meant to be individually included. Include c10/util/complex.h inst…
11 C10_HOST_DEVICE inline c10::complex<T> exp(const c10::complex<T>& x) { in exp()
13 return static_cast<c10::complex<T>>( in exp()
14 thrust::exp(static_cast<thrust::complex<T>>(x))); in exp()
16 return static_cast<c10::complex<T>>( in exp()
17 std::exp(static_cast<std::complex<T>>(x))); in exp()
22 C10_HOST_DEVICE inline c10::complex<T> log(const c10::complex<T>& x) { in log()
24 return static_cast<c10::complex<T>>( in log()
25 thrust::log(static_cast<thrust::complex<T>>(x))); in log()
27 return static_cast<c10::complex<T>>( in log()
28 std::log(static_cast<std::complex<T>>(x))); in log()
33 C10_HOST_DEVICE inline c10::complex<T> log10(const c10::complex<T>& x) { in log10()
35 return static_cast<c10::complex<T>>( in log10()
36 thrust::log10(static_cast<thrust::complex<T>>(x))); in log10()
38 return static_cast<c10::complex<T>>( in log10()
39 std::log10(static_cast<std::complex<T>>(x))); in log10()
44 C10_HOST_DEVICE inline c10::complex<T> log2(const c10::complex<T>& x) { in log2()
45 const c10::complex<T> log2 = c10::complex<T>(::log(2.0), 0.0); in log2()
54 C10_API c10::complex<float> sqrt(const c10::complex<float>& in);
55 C10_API c10::complex<double> sqrt(const c10::complex<double>& in);
56 C10_API c10::complex<float> acos(const c10::complex<float>& in);
57 C10_API c10::complex<double> acos(const c10::complex<double>& in);
62 C10_HOST_DEVICE inline c10::complex<T> sqrt(const c10::complex<T>& x) { in sqrt()
64 return static_cast<c10::complex<T>>( in sqrt()
65 thrust::sqrt(static_cast<thrust::complex<T>>(x))); in sqrt()
69 return static_cast<c10::complex<T>>( in sqrt()
70 std::sqrt(static_cast<std::complex<T>>(x))); in sqrt()
77 C10_HOST_DEVICE inline c10::complex<T> pow( in pow()
78 const c10::complex<T>& x, in pow()
79 const c10::complex<T>& y) { in pow()
81 return static_cast<c10::complex<T>>(thrust::pow( in pow()
82 static_cast<thrust::complex<T>>(x), static_cast<thrust::complex<T>>(y))); in pow()
84 return static_cast<c10::complex<T>>(std::pow( in pow()
85 static_cast<std::complex<T>>(x), static_cast<std::complex<T>>(y))); in pow()
90 C10_HOST_DEVICE inline c10::complex<T> pow( in pow()
91 const c10::complex<T>& x, in pow()
94 return static_cast<c10::complex<T>>( in pow()
95 thrust::pow(static_cast<thrust::complex<T>>(x), y)); in pow()
97 return static_cast<c10::complex<T>>( in pow()
98 std::pow(static_cast<std::complex<T>>(x), y)); in pow()
103 C10_HOST_DEVICE inline c10::complex<T> pow( in pow()
105 const c10::complex<T>& y) { in pow()
107 return static_cast<c10::complex<T>>( in pow()
108 thrust::pow(x, static_cast<thrust::complex<T>>(y))); in pow()
110 return static_cast<c10::complex<T>>( in pow()
111 std::pow(x, static_cast<std::complex<T>>(y))); in pow()
116 C10_HOST_DEVICE inline c10::complex<decltype(T() * U())> pow( in pow()
117 const c10::complex<T>& x, in pow()
118 const c10::complex<U>& y) { in pow()
120 return static_cast<c10::complex<T>>(thrust::pow( in pow()
121 static_cast<thrust::complex<T>>(x), static_cast<thrust::complex<T>>(y))); in pow()
123 return static_cast<c10::complex<T>>(std::pow( in pow()
124 static_cast<std::complex<T>>(x), static_cast<std::complex<T>>(y))); in pow()
129 C10_HOST_DEVICE inline c10::complex<decltype(T() * U())> pow( in pow()
130 const c10::complex<T>& x, in pow()
133 return static_cast<c10::complex<T>>( in pow()
134 thrust::pow(static_cast<thrust::complex<T>>(x), y)); in pow()
136 return static_cast<c10::complex<T>>( in pow()
137 std::pow(static_cast<std::complex<T>>(x), y)); in pow()
142 C10_HOST_DEVICE inline c10::complex<decltype(T() * U())> pow( in pow()
144 const c10::complex<U>& y) { in pow()
146 return static_cast<c10::complex<T>>( in pow()
147 thrust::pow(x, static_cast<thrust::complex<T>>(y))); in pow()
149 return static_cast<c10::complex<T>>( in pow()
150 std::pow(x, static_cast<std::complex<T>>(y))); in pow()
157 C10_HOST_DEVICE inline c10::complex<T> sin(const c10::complex<T>& x) { in sin()
159 return static_cast<c10::complex<T>>( in sin()
160 thrust::sin(static_cast<thrust::complex<T>>(x))); in sin()
162 return static_cast<c10::complex<T>>( in sin()
163 std::sin(static_cast<std::complex<T>>(x))); in sin()
168 C10_HOST_DEVICE inline c10::complex<T> cos(const c10::complex<T>& x) { in cos()
170 return static_cast<c10::complex<T>>( in cos()
171 thrust::cos(static_cast<thrust::complex<T>>(x))); in cos()
173 return static_cast<c10::complex<T>>( in cos()
174 std::cos(static_cast<std::complex<T>>(x))); in cos()
179 C10_HOST_DEVICE inline c10::complex<T> tan(const c10::complex<T>& x) { in tan()
181 return static_cast<c10::complex<T>>( in tan()
182 thrust::tan(static_cast<thrust::complex<T>>(x))); in tan()
184 return static_cast<c10::complex<T>>( in tan()
185 std::tan(static_cast<std::complex<T>>(x))); in tan()
190 C10_HOST_DEVICE inline c10::complex<T> asin(const c10::complex<T>& x) { in asin()
192 return static_cast<c10::complex<T>>( in asin()
193 thrust::asin(static_cast<thrust::complex<T>>(x))); in asin()
195 return static_cast<c10::complex<T>>( in asin()
196 std::asin(static_cast<std::complex<T>>(x))); in asin()
201 C10_HOST_DEVICE inline c10::complex<T> acos(const c10::complex<T>& x) { in acos()
203 return static_cast<c10::complex<T>>( in acos()
204 thrust::acos(static_cast<thrust::complex<T>>(x))); in acos()
206 return static_cast<c10::complex<T>>( in acos()
207 std::acos(static_cast<std::complex<T>>(x))); in acos()
214 C10_HOST_DEVICE inline c10::complex<T> atan(const c10::complex<T>& x) { in atan()
216 return static_cast<c10::complex<T>>( in atan()
217 thrust::atan(static_cast<thrust::complex<T>>(x))); in atan()
219 return static_cast<c10::complex<T>>( in atan()
220 std::atan(static_cast<std::complex<T>>(x))); in atan()
227 C10_HOST_DEVICE inline c10::complex<T> sinh(const c10::complex<T>& x) { in sinh()
229 return static_cast<c10::complex<T>>( in sinh()
230 thrust::sinh(static_cast<thrust::complex<T>>(x))); in sinh()
232 return static_cast<c10::complex<T>>( in sinh()
233 std::sinh(static_cast<std::complex<T>>(x))); in sinh()
238 C10_HOST_DEVICE inline c10::complex<T> cosh(const c10::complex<T>& x) { in cosh()
240 return static_cast<c10::complex<T>>( in cosh()
241 thrust::cosh(static_cast<thrust::complex<T>>(x))); in cosh()
243 return static_cast<c10::complex<T>>( in cosh()
244 std::cosh(static_cast<std::complex<T>>(x))); in cosh()
249 C10_HOST_DEVICE inline c10::complex<T> tanh(const c10::complex<T>& x) { in tanh()
251 return static_cast<c10::complex<T>>( in tanh()
252 thrust::tanh(static_cast<thrust::complex<T>>(x))); in tanh()
254 return static_cast<c10::complex<T>>( in tanh()
255 std::tanh(static_cast<std::complex<T>>(x))); in tanh()
260 C10_HOST_DEVICE inline c10::complex<T> asinh(const c10::complex<T>& x) { in asinh()
262 return static_cast<c10::complex<T>>( in asinh()
263 thrust::asinh(static_cast<thrust::complex<T>>(x))); in asinh()
265 return static_cast<c10::complex<T>>( in asinh()
266 std::asinh(static_cast<std::complex<T>>(x))); in asinh()
271 C10_HOST_DEVICE inline c10::complex<T> acosh(const c10::complex<T>& x) { in acosh()
273 return static_cast<c10::complex<T>>( in acosh()
274 thrust::acosh(static_cast<thrust::complex<T>>(x))); in acosh()
276 return static_cast<c10::complex<T>>( in acosh()
277 std::acosh(static_cast<std::complex<T>>(x))); in acosh()
282 C10_HOST_DEVICE inline c10::complex<T> atanh(const c10::complex<T>& x) { in atanh()
284 return static_cast<c10::complex<T>>( in atanh()
285 thrust::atanh(static_cast<thrust::complex<T>>(x))); in atanh()
287 return static_cast<c10::complex<T>>( in atanh()
288 std::atanh(static_cast<std::complex<T>>(x))); in atanh()
293 C10_HOST_DEVICE inline c10::complex<T> log1p(const c10::complex<T>& z) { in log1p()
299 // For CUDA we also use this one, as thrust::log(thrust::complex) takes in log1p()
330 c10::complex<T> u = z + T(1); in log1p()
344 C10_HOST_DEVICE inline c10::complex<T> expm1(const c10::complex<T>& z) { in expm1()