Searched refs:complex_input (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | CuFFTPlanCache.h | 90 inline CuFFTTransformType GetCuFFTTransformType(bool complex_input, bool complex_output) { in GetCuFFTTransformType() argument
91 if (complex_input && complex_output) { in GetCuFFTTransformType()
93 } else if (complex_input && !complex_output) { in GetCuFFTTransformType()
95 } else if (!complex_input && complex_output) { in GetCuFFTTransformType()
299 const auto complex_input = cufft_complex_input(fft_type); in CuFFTConfig() local
302 itype = complex_input ? CUDA_C_32F : CUDA_R_32F; in CuFFTConfig()
306 itype = complex_input ? CUDA_C_64F : CUDA_R_64F; in CuFFTConfig()
310 itype = complex_input ? CUDA_C_16F : CUDA_R_16F; in CuFFTConfig()
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_ops_test.h | 247 absl::InlinedVector<T, 10> complex_input; in DefaultInput() local
249 complex_input.emplace_back(value, -value); in DefaultInput()
251 return complex_input; in DefaultInput()
263 absl::InlinedVector<T, 10> complex_input; in ComplexInputFromValues() local
266 complex_input.emplace_back(real[i], imag[i]); in ComplexInputFromValues()
268 return complex_input; in ComplexInputFromValues()
|
| /external/pytorch/aten/src/ATen/native/mkl/ |
| D | SpectralOps.cpp | 351 bool complex_input, bool complex_output, in _plan_mkl_fft() argument
368 return complex_input ? DFTI_COMPLEX : DFTI_REAL; in _plan_mkl_fft()
403 if (!complex_input || !complex_output) { in _plan_mkl_fft()
|
| /external/pytorch/torch/csrc/autograd/ |
| D | FunctionsManual.h | 730 bool complex_input,
|
| /external/pytorch/test/inductor/ |
| D | test_torchinductor.py | 1120 complex_input = torch.tensor(
1125 self.common(fn, (complex_input, real_input, interger_real_input))
|