Searched refs:complex_to_real (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | Convolution.cpp | 837 std::pair<Tensor, Tensor> complex_to_real(const Tensor& inp) { in complex_to_real() function
856 auto [i_r, i_i] = complex_to_real(input.resolve_conj()); in complex_convolution()
857 auto [w_r, w_i] = complex_to_real(weight.resolve_conj()); in complex_convolution()
873 auto [b_r, b_i] = complex_to_real(bias.resolve_conj()); in complex_convolution()
893 auto [i_r, i_i] = complex_to_real(input.resolve_conj()); in complex_convolution_mode()
894 auto [w_r, w_i] = complex_to_real(weight.resolve_conj()); in complex_convolution_mode()
903 auto [b_r, b_i] = complex_to_real(bias.resolve_conj()); in complex_convolution_mode()
|
| /external/pytorch/torch/testing/_internal/opinfo/ |
| D | core.py | 1581 complex_to_real: bool = False,
1599 assert not (result_dtype and complex_to_real)
1619 self.complex_to_real = complex_to_real
|
| /external/pytorch/torch/testing/_internal/opinfo/definitions/ |
| D | linalg.py | 1849 complex_to_real=True,
|
| /external/pytorch/torch/testing/_internal/ |
| D | common_methods_invocations.py | 20807 complex_to_real=True,
20836 complex_to_real=True,
20857 complex_to_real=True,
20884 complex_to_real=True,
|
| /external/pytorch/test/ |
| D | test_reductions.py | 265 elif op.complex_to_real:
|