Lines Matching defs:dim
162 IntArrayRef dim, int64_t norm, bool onesided) { in fft_r2c_maybe_out()
174 IntArrayRef dim, int64_t norm, SymInt last_dim_size) { in fft_c2r_maybe_out()
188 IntArrayRef dim, int64_t norm, bool forward) { in fft_c2c_maybe_out()
207 const auto dim = maybe_wrap_dim(unwrapped_dim, input_dim, /*wrap_scalar=*/false); in fft_c2r() local
233 const auto dim = maybe_wrap_dim(unwrapped_dim, input_dim, /*wrap_scalar=*/false); in fft_r2c() local
265 const auto dim = maybe_wrap_dim(unwrapped_dim, input_dim, /*wrap_scalar=*/false); in fft_c2c() local
278 DimVector dim; member
285 Tensor input, at::OptionalSymIntArrayRef shape, at::OptionalIntArrayRef dim) { in canonicalize_fft_shape_and_dim_args()
349 IntArrayRef dim, std::optional<c10::string_view> norm_str, bool forward) { in fftn_c2c()
360 Tensor fft_fft_symint(const Tensor& self, std::optional<SymInt> n, int64_t dim, in fft_fft_symint()
368 int64_t dim, std::optional<c10::string_view> norm, Tensor& out) { in fft_fft_symint_out()
377 Tensor fft_ifft_symint(const Tensor& self, std::optional<SymInt> n, int64_t dim, in fft_ifft_symint()
385 int64_t dim, std::optional<c10::string_view> norm, Tensor& out) { in fft_ifft_symint_out()
394 Tensor fft_rfft_symint(const Tensor& self, std::optional<SymInt> n, int64_t dim, in fft_rfft_symint()
400 int64_t dim, std::optional<c10::string_view> norm, Tensor& out) { in fft_rfft_symint_out()
405 Tensor fft_irfft_symint(const Tensor& self, std::optional<SymInt> n, int64_t dim, in fft_irfft_symint()
411 int64_t dim, std::optional<c10::string_view> norm, Tensor& out) { in fft_irfft_symint_out()
416 Tensor fft_hfft_symint(const Tensor& self, std::optional<SymInt> n, int64_t dim, in fft_hfft_symint()
422 int64_t dim, std::optional<c10::string_view> norm, Tensor& out) { in fft_hfft_symint_out()
427 Tensor fft_ihfft_symint(const Tensor& self, std::optional<SymInt> n, int64_t dim, in fft_ihfft_symint()
433 int64_t dim, std::optional<c10::string_view> norm, Tensor& out) { in fft_ihfft_symint_out()
439 at::OptionalIntArrayRef dim, in fft_fftn_symint()
449 at::OptionalIntArrayRef dim, in fft_fftn_symint_out()
459 at::OptionalIntArrayRef dim, in fft_ifftn_symint()
468 at::OptionalIntArrayRef dim, in fft_ifftn_symint_out()
478 at::OptionalIntArrayRef dim, in fft_rfftn_impl()
491 at::OptionalIntArrayRef dim, in fft_rfftn_symint()
498 at::OptionalIntArrayRef dim, in fft_rfftn_symint_out()
530 at::OptionalIntArrayRef dim, in fft_irfftn_impl()
544 at::OptionalIntArrayRef dim, in fft_irfftn_symint()
551 at::OptionalIntArrayRef dim, in fft_irfftn_symint_out()
560 at::OptionalIntArrayRef dim, in fft_hfftn_impl()
588 at::OptionalIntArrayRef dim, in fft_hfftn_symint()
596 at::OptionalIntArrayRef dim, std::optional<c10::string_view> norm, in fft_hfftn_symint_out()
605 const at::OptionalIntArrayRef& dim, in fft_ihfftn_impl()
630 at::OptionalIntArrayRef dim, in fft_ihfftn_symint()
638 at::OptionalIntArrayRef dim, in fft_ihfftn_symint_out()
646 IntArrayRef dim, std::optional<c10::string_view> norm) { in fft_fft2_symint()
651 IntArrayRef dim, std::optional<c10::string_view> norm, Tensor& out) { in fft_fft2_symint_out()
656 IntArrayRef dim, std::optional<c10::string_view> norm) { in fft_ifft2_symint()
661 IntArrayRef dim, std::optional<c10::string_view> norm, Tensor& out) { in fft_ifft2_symint_out()
666 IntArrayRef dim, std::optional<c10::string_view> norm) { in fft_rfft2_symint()
671 IntArrayRef dim, std::optional<c10::string_view> norm, Tensor& out) { in fft_rfft2_symint_out()
676 IntArrayRef dim, std::optional<c10::string_view> norm) { in fft_irfft2_symint()
681 IntArrayRef dim, std::optional<c10::string_view> norm, Tensor& out) { in fft_irfft2_symint_out()
686 const Tensor& self, at::OptionalSymIntArrayRef s, IntArrayRef dim, in fft_hfft2_symint_out()
692 IntArrayRef dim, std::optional<c10::string_view> norm) { in fft_hfft2_symint()
697 const Tensor& self, at::OptionalSymIntArrayRef s, IntArrayRef dim, in fft_ihfft2_symint_out()
703 IntArrayRef dim, std::optional<c10::string_view> norm) { in fft_ihfft2_symint()
754 DimVector dim; in default_alldims() local
769 auto dim = default_alldims(x, dim_opt); in fft_fftshift() local
781 auto dim = default_alldims(x, dim_opt); in fft_ifftshift() local
1202 DimVector dim(dim_.begin(), dim_.end()); in _fft_fill_with_conjugate_symmetry_() local
1211 std::remove_if(dim.begin(), dim.end(), [&](int64_t dim) { in _fft_fill_with_conjugate_symmetry_()