Searched refs:OptionalSymIntArrayRef (Results 1 – 11 of 11) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
H A D | SpectralOps.cpp | 285 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 D | ConvUtils.h | 127 …ed, SymIntArrayRef output_padding, c10::SymInt groups, const at::OptionalSymIntArrayRef bias_sizes…
|
H A D | Convolution.cpp | 270 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 D | KernelFunction.h | 24 std::is_same<at::OptionalSymIntArrayRef, T>, 39 struct remove_symint<at::OptionalSymIntArrayRef> {
|
H A D | KernelFunction_impl.h | 79 inline typename remove_symint<at::OptionalSymIntArrayRef>::type unpackSymInt(at::OptionalSymIntArra… in unpackSymInt()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
H A D | ATen_fwd.h | 36 using OptionalSymIntArrayRef = c10::OptionalArrayRef<c10::SymInt>; variable
|
H A D | function_schema.cpp | 35 *a.real_type() == *getTypePtr<at::OptionalSymIntArrayRef>() in cloneWithRealTypes()
|
H A D | jit_type.h | 2066 struct getMaybeFakeTypePtr_<at::OptionalSymIntArrayRef, fake> final {
|
/aosp_15_r20/external/pytorch/test/inductor/ |
H A D | custom_ops.cpp | 251 at::OptionalSymIntArrayRef o_symints, in fn_with_all_inputs_meta()
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/ |
H A D | tracer.h | 275 const at::OptionalSymIntArrayRef& opt_value);
|
H A D | tracer.cpp | 848 const at::OptionalSymIntArrayRef& opt_value) { in addInputs()
|