Searched full:fft_irfftn (Results 1 – 12 of 12) sorted by relevance
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | testing.py | 147 aten.fft_irfftn.default,
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | annotated_ops | 519 fft_irfftn, fft
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesDecompositions.cpp | 122 m.impl("fft_irfftn", native::fft_irfftn_symint); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | model_ops.yaml | 144 aten::fft_irfftn: 1
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | amp.rst | 462 ``fft_irfftn``,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | autocast_mode.cpp | 419 KERNEL_CPU(fft_irfftn, fp32) in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/ |
| H A D | fft.h | 201 return torch::fft_irfftn(self, s, dim, norm);
|
| /aosp_15_r20/external/pytorch/torch/_refs/ |
| H A D | fft.py | 465 @register_decomposition(aten.fft_irfftn)
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/definitions/ |
| H A D | fft.py | 588 aten_name="fft_irfftn",
|
| /aosp_15_r20/external/pytorch/torch/fft/ |
| H A D | __init__.py | 655 irfftn = _add_docstr(_fft.fft_irfftn, r"""
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | trace_rules.py | 567 "torch._C._fft.fft_irfftn",
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | native_functions.yaml | 13675 - func: fft_irfftn(Tensor self, SymInt[1]? s=None, int[1]? dim=None, str? norm=None) -> Tensor
13681 - func: fft_irfftn.out(Tensor self, SymInt[1]? s=None, int[1]? dim=None, str? norm=None, *, Tensor(…
|