Home
last modified time | relevance | path

Searched refs:OptionalSymIntArrayRef (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DSpectralOps.cpp285 Tensor input, at::OptionalSymIntArrayRef shape, at::OptionalIntArrayRef dim) { in canonicalize_fft_shape_and_dim_args()
438 Tensor fft_fftn_symint(const Tensor& self, at::OptionalSymIntArrayRef s, in fft_fftn_symint()
448 at::OptionalSymIntArrayRef s, in fft_fftn_symint_out()
458 Tensor fft_ifftn_symint(const Tensor& self, at::OptionalSymIntArrayRef s, in fft_ifftn_symint()
467 at::OptionalSymIntArrayRef s, in fft_ifftn_symint_out()
477 at::OptionalSymIntArrayRef s, in fft_rfftn_impl()
490 Tensor fft_rfftn_symint(const Tensor& self, at::OptionalSymIntArrayRef s, in fft_rfftn_symint()
497 at::OptionalSymIntArrayRef s, in fft_rfftn_symint_out()
506 const at::OptionalSymIntArrayRef& s, in canonicalize_fft_c2r_shape_and_dim_args()
529 at::OptionalSymIntArrayRef s, in fft_irfftn_impl()
[all …]
H A DConvUtils.h127 …ed, SymIntArrayRef output_padding, c10::SymInt groups, const at::OptionalSymIntArrayRef bias_sizes…
H A DConvolution.cpp270 const at::OptionalSymIntArrayRef bias_sizes_opt, in xnnpack_use_convolution2d()
1314 …, SymIntArrayRef output_padding_, c10::SymInt groups_, const at::OptionalSymIntArrayRef bias_sizes… in select_conv_backend()
/aosp_15_r20/external/pytorch/aten/src/ATen/core/boxing/
H A DKernelFunction.h24 std::is_same<at::OptionalSymIntArrayRef, T>,
39 struct remove_symint<at::OptionalSymIntArrayRef> {
H A DKernelFunction_impl.h79 inline typename remove_symint<at::OptionalSymIntArrayRef>::type unpackSymInt(at::OptionalSymIntArra… in unpackSymInt()
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DATen_fwd.h36 using OptionalSymIntArrayRef = c10::OptionalArrayRef<c10::SymInt>; variable
H A Dfunction_schema.cpp35 *a.real_type() == *getTypePtr<at::OptionalSymIntArrayRef>() in cloneWithRealTypes()
H A Djit_type.h2066 struct getMaybeFakeTypePtr_<at::OptionalSymIntArrayRef, fake> final {
/aosp_15_r20/external/pytorch/test/inductor/
H A Dcustom_ops.cpp251 at::OptionalSymIntArrayRef o_symints, in fn_with_all_inputs_meta()
/aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/
H A Dtracer.h275 const at::OptionalSymIntArrayRef& opt_value);
H A Dtracer.cpp848 const at::OptionalSymIntArrayRef& opt_value) { in addInputs()