Home
last modified time | relevance | path

Searched defs:opt_dim (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/
H A Dlinalg.h133 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()
580 OptionalIntArrayRef opt_dim, in linalg_norm()
590 OptionalIntArrayRef opt_dim, in linalg_norm()
602 OptionalIntArrayRef opt_dim, in linalg_norm_out()
614 OptionalIntArrayRef opt_dim, in linalg_norm_out()
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DReduceOps.cpp32 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()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DVar.cpp15 const OptionalIntArrayRef opt_dim, in var_dim_IntList()
H A DMean.cpp103 const OptionalIntArrayRef opt_dim, in mean_dim_IntList()
H A DSum.cpp92 const OptionalIntArrayRef opt_dim, in sum_dim_IntList()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLinearAlgebra.cpp2995 …lg_norm(const Tensor& X, const std::optional<Scalar>& opt_ord, OptionalIntArrayRef opt_dim, bool k… in linalg_norm()
3020 …orm_out(const Tensor& X, const std::optional<Scalar>& opt_ord, OptionalIntArrayRef opt_dim, bool k… in linalg_norm_out()
3032 Tensor linalg_norm(const Tensor& X, c10::string_view ord, OptionalIntArrayRef opt_dim, bool keepdim… in linalg_norm()
3045 Tensor& linalg_norm_out(const Tensor& X, c10::string_view ord, OptionalIntArrayRef opt_dim, bool ke… in linalg_norm_out()
H A DTensorAdvancedIndexing.cpp2202 Tensor take_along_dim(const Tensor& self, const Tensor& indices, std::optional<int64_t> opt_dim) { in take_along_dim()
2214 …ake_along_dim_out(const Tensor& self, const Tensor& indices, std::optional<int64_t> opt_dim, Tenso… in take_along_dim_out()
H A DReduceOps.cpp2059 Tensor& std_out(const Tensor& self, at::OptionalIntArrayRef opt_dim, bool unbiased, bool keepdim, T… in std_out()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DFunctionsManual.cpp446 const at::OptionalIntArrayRef& opt_dim, in linalg_vector_norm_jvp()
459 const at::OptionalIntArrayRef& opt_dim, in linalg_vector_norm_backward()
684 OptionalIntArrayRef opt_dim, in unsqueeze_multiple()
747 OptionalIntArrayRef opt_dim, in mean_backward()
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Deval.cpp713 auto opt_dim = intValue(dims[i]); in check_bounds() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
H A DQuantizedOpKernels.cpp2926 OptionalIntArrayRef opt_dim,