Searched full:fft_rfftn (Results 1 – 12 of 12) sorted by relevance
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | testing.py | 145 aten.fft_rfftn.default,
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | annotated_ops | 518 fft_rfftn, fft
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesDecompositions.cpp | 125 m.impl("fft_rfftn", native::fft_rfftn_symint); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | model_ops.yaml | 145 aten::fft_rfftn: 1
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | amp.rst | 461 ``fft_rfftn``,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | autocast_mode.cpp | 418 KERNEL_CPU(fft_rfftn, fp32) in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/ |
| H A D | fft.h | 185 return torch::fft_rfftn(self, s, dim, norm);
|
| /aosp_15_r20/external/pytorch/torch/_refs/ |
| H A D | fft.py | 384 @register_decomposition(aten.fft_rfftn)
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/definitions/ |
| H A D | fft.py | 354 aten_name="fft_rfftn",
|
| /aosp_15_r20/external/pytorch/torch/fft/ |
| H A D | __init__.py | 591 rfftn = _add_docstr(_fft.fft_rfftn, r"""
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | trace_rules.py | 571 "torch._C._fft.fft_rfftn",
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | native_functions.yaml | 13663 - func: fft_rfftn(Tensor self, SymInt[1]? s=None, int[1]? dim=None, str? norm=None) -> Tensor
13669 - func: fft_rfftn.out(Tensor self, SymInt[1]? s=None, int[1]? dim=None, str? norm=None, *, Tensor(a…
|