| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/ |
| H A D | fft.h | 50 OptionalIntArrayRef s = std::nullopt,
66 at::OptionalIntArrayRef s = std::nullopt,
82 at::OptionalIntArrayRef s = std::nullopt,
83 at::OptionalIntArrayRef dim = std::nullopt,
98 at::OptionalIntArrayRef s = std::nullopt,
99 at::OptionalIntArrayRef dim = std::nullopt,
150 at::OptionalIntArrayRef s = std::nullopt,
166 at::OptionalIntArrayRef s = std::nullopt,
182 at::OptionalIntArrayRef s = std::nullopt,
183 at::OptionalIntArrayRef dim = std::nullopt,
[all …]
|
| H A D | linalg.h | 133 OptionalIntArrayRef opt_dim, in norm()
142 OptionalIntArrayRef opt_dim, in norm()
152 OptionalIntArrayRef opt_dim, in norm_out()
163 OptionalIntArrayRef opt_dim, in norm_out()
172 OptionalIntArrayRef opt_dim, in vector_norm()
182 OptionalIntArrayRef opt_dim, in vector_norm_out()
423 OptionalIntArrayRef dims) { in tensorsolve()
431 OptionalIntArrayRef dims) { in tensorsolve_out()
580 OptionalIntArrayRef opt_dim, in linalg_norm()
590 OptionalIntArrayRef opt_dim, in linalg_norm()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ReduceOps.cpp | 185 OptionalIntArrayRef dims, in allany_meta()
197 TORCH_META_FUNC2(all, dims)(const Tensor& self, OptionalIntArrayRef dim, bool keepdim) { in TORCH_META_FUNC2()
209 TORCH_META_FUNC2(any, dims)(const Tensor& self, OptionalIntArrayRef dim, bool keepdim) { in TORCH_META_FUNC2()
277 (const Tensor& self, OptionalIntArrayRef opt_dim, bool keepdim, std::optional<ScalarType> opt_dtype… in TORCH_META_FUNC2()
292 (const Tensor& self, OptionalIntArrayRef opt_dim, bool keepdim, std::optional<ScalarType> opt_dtype… in TORCH_META_FUNC2()
1003 …dient(const Tensor& self, std::optional<int64_t> spacing_size, at::OptionalIntArrayRef dim, int64… in pre_check_gradient()
1110 at::OptionalIntArrayRef(dim), in gradient()
1119 dim.has_value() ? at::OptionalIntArrayRef(processed_dim) : std::nullopt, in gradient()
1127 at::OptionalIntArrayRef(dim), in gradient()
1136 dim.has_value() ? at::OptionalIntArrayRef(processed_dim) : std::nullopt, in gradient()
[all …]
|
| H A D | ReduceOpsUtils.h | 114 inline DimVector make_dim_vector(OptionalIntArrayRef opt_dims, int64_t ndim) { in make_dim_vector()
124 inline DimMask make_dim_mask(OptionalIntArrayRef opt_dims, int64_t ndim, bool allow_empty_dims=fals…
163 const Tensor& self, at::OptionalIntArrayRef dim, bool keepdim, ScalarType dtype in create_reduction_result()
187 at::OptionalIntArrayRef dim_opt, in make_reduction()
212 at::OptionalIntArrayRef dim, bool keepdim, ScalarType out_dtype) { in make_reduction()
227 at::OptionalIntArrayRef dim_opt, bool keepdim, ScalarType dtype1, in make_reduction()
264 at::OptionalIntArrayRef dim, bool keepdim, ScalarType dtype) { in make_reduction()
364 OptionalIntArrayRef opt_dims,
402 OptionalIntArrayRef opt_dims, in make_reduction()
440 OptionalIntArrayRef opt_dims, in make_reduction_from_out_ty()
|
| H A D | SpectralOps.cpp | 285 Tensor input, at::OptionalSymIntArrayRef shape, at::OptionalIntArrayRef dim) { in canonicalize_fft_shape_and_dim_args()
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()
507 const at::OptionalIntArrayRef& dims, in canonicalize_fft_c2r_shape_and_dim_args()
530 at::OptionalIntArrayRef dim, in fft_irfftn_impl()
[all …]
|
| H A D | ComparisonUtils.cpp | 27 …sert_tensor_metadata(at::Tensor const& tensor, at::OptionalIntArrayRef sizes, at::OptionalIntArray… in _assert_tensor_metadata()
|
| H A D | UpSampleNearest1d.cpp | 127 at::OptionalIntArrayRef output_size, in upsample_nearest1d()
136 at::OptionalIntArrayRef output_size, in _upsample_nearest_exact1d()
|
| H A D | UpSampleNearest2d.cpp | 151 at::OptionalIntArrayRef output_size, in upsample_nearest2d()
161 at::OptionalIntArrayRef output_size, in _upsample_nearest_exact2d()
|
| H A D | UpSampleNearest3d.cpp | 166 at::OptionalIntArrayRef output_size, in upsample_nearest3d()
177 at::OptionalIntArrayRef output_size, in _upsample_nearest_exact3d()
|
| H A D | UpSampleBilinear2d.cpp | 162 at::OptionalIntArrayRef output_size, in upsample_bilinear2d()
173 at::OptionalIntArrayRef output_size, in _upsample_bilinear2d_aa()
|
| H A D | UpSampleBicubic2d.cpp | 281 at::OptionalIntArrayRef output_size, in upsample_bicubic2d()
292 at::OptionalIntArrayRef output_size, in _upsample_bicubic2d_aa()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | ReduceOps.cpp | 32 OptionalIntArrayRef opt_dim) { in is_innnermost_dim()
49 OptionalIntArrayRef opt_dim, in is_mean_inner_dim_fast_path()
132 OptionalIntArrayRef opt_dim, in mean_out_quantized_cpu()
167 OptionalIntArrayRef opt_dim, in mean_quantized_cpu()
178 OptionalIntArrayRef dim, in is_std_inner_dim_fast_path()
198 OptionalIntArrayRef dim, in std_out_quantized_cpu()
222 OptionalIntArrayRef dim, in std_quantized_cpu()
|
| H A D | QuantizedOps.h | 193 OptionalIntArrayRef /* opt_dim */,
200 OptionalIntArrayRef /* dim */,
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.h | 113 const at::OptionalIntArrayRef& opt_dim,
120 const at::OptionalIntArrayRef& opt_dim,
172 at::OptionalIntArrayRef opt_dim,
177 at::OptionalIntArrayRef opt_dims,
187 at::OptionalIntArrayRef dims,
378 at::OptionalIntArrayRef dim,
385 at::OptionalIntArrayRef dim_opt,
392 at::OptionalIntArrayRef dim,
398 at::OptionalIntArrayRef opt_dim,
405 at::OptionalIntArrayRef dim_opt,
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | TensorShapeCUDA.cpp | 35 at::OptionalIntArrayRef stride_opt = stride.data() != nullptr ? in set_storage_cuda_()
36 at::OptionalIntArrayRef(stride) : std::nullopt; in set_storage_cuda_()
|
| /aosp_15_r20/external/pytorch/c10/util/ |
| H A D | OptionalArrayRef.h | 219 using OptionalIntArrayRef = OptionalArrayRef<int64_t>; variable
222 const OptionalIntArrayRef& a1,
232 const c10::OptionalIntArrayRef& a2) {
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/ |
| H A D | Convolution.cpp | 29 const at::OptionalIntArrayRef bias_sizes_opt, in available()
191 … (bias.has_value() && bias->defined()) ? at::OptionalIntArrayRef(bias->sizes()) : std::nullopt, in create()
454 (bias && bias->defined()) ? at::OptionalIntArrayRef(bias->sizes()) : std::nullopt, in unpack_prepacked_sizes_conv2d()
473 const at::OptionalIntArrayRef bias_sizes_opt, in use_convolution2d()
|
| H A D | Shim.cpp | 34 const at::OptionalIntArrayRef, in use_convolution2d() argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/ |
| H A D | ReduceOps.mm | 104 OptionalIntArrayRef opt_dim,
121 OptionalIntArrayRef opt_dims,
148 OptionalIntArrayRef opt_dim,
428 at::OptionalIntArrayRef dim,
898 OptionalIntArrayRef opt_dim,
906 OptionalIntArrayRef dim,
921 Tensor nansum_mps(const Tensor& self, OptionalIntArrayRef dim, bool keepdim, std::optional<ScalarTy…
964 dim_opt.has_value() ? OptionalIntArrayRef({*dim_opt}) : std::nullopt,
971 dim_opt.has_value() ? OptionalIntArrayRef({*dim_opt}) : std::nullopt,
1023 OptionalIntArrayRef opt_dim,
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ops/ |
| H A D | from_blob.h | 28 TensorMaker& strides(OptionalIntArrayRef value) noexcept { in strides()
93 OptionalIntArrayRef strides_{};
|
| /aosp_15_r20/external/executorch/runtime/core/exec_aten/ |
| H A D | exec_aten.h | 91 using OptionalIntArrayRef = OptionalArrayRef<int64_t>; variable
137 using OptionalIntArrayRef = OptionalArrayRef<int64_t>; variable
|
| /aosp_15_r20/external/pytorch/test/cpp_extensions/ |
| H A D | open_registration_extension.cpp | 400 …at::OptionalIntArrayRef stride_opt = stride.data() != nullptr ? at::OptionalIntArrayRef(stride) : … in custom_set_source_Storage()
414 …at::OptionalIntArrayRef stride_opt = stride.data() != nullptr ? at::OptionalIntArrayRef(stride) : … in custom_set_source_Storage_storage_offset()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorAliases.cpp | 10 OptionalIntArrayRef output_size) { in nested_to_padded_tensor()
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | shape_inference.h | 37 …d_output, const at::Tensor & input, const at::Tensor & weight, at::OptionalIntArrayRef bias_sizes,…
84 …torch::lazy::Shape> compute_shape_std(const at::Tensor & self, at::OptionalIntArrayRef dim, bool u…
85 …torch::lazy::Shape> compute_shape_std(const at::Tensor & self, at::OptionalIntArrayRef dim, const …
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | WrapDimUtilsMulti.h | 17 OptionalIntArrayRef opt_dims, in dim_list_to_bitset()
|