Lines Matching defs:dim
193 TORCH_META_FUNC2(all, dim)(const Tensor& self, int64_t dim, bool keepdim) { in TORCH_META_FUNC2() argument
205 TORCH_META_FUNC2(any, dim)(const Tensor& self, int64_t dim, bool keepdim) { in TORCH_META_FUNC2() argument
220 const std::optional<int64_t>& dim) { in check_argmax_argmin()
247 int64_t dim, in meta_func_cum_ops()
291 TORCH_META_FUNC2(mean, dim) in TORCH_META_FUNC2() argument
356 auto dim = maybe_wrap_dim(dim_opt.value(), self.ndimension()); in TORCH_META_FUNC() local
445 Tensor _logcumsumexp_cpu(const Tensor& self, int64_t dim) { in _logcumsumexp_cpu()
450 Tensor& _logcumsumexp_out_cpu(const Tensor& self, int64_t dim, Tensor& result) { in _logcumsumexp_out_cpu()
455 Tensor logcumsumexp(const Tensor& self, int64_t dim) { in logcumsumexp()
464 Tensor& logcumsumexp_out(const Tensor& self, int64_t dim, Tensor& result) { in logcumsumexp_out()
477 int64_t dim, in impl_func_cum_ops()
507 static Tensor reversed_cumsum(const Tensor& w, int64_t dim) { in reversed_cumsum()
511 Tensor cumprod_backward(const Tensor& grad, const Tensor& input, int64_t dim, const Tensor& output)… in cumprod_backward()
788 void cummax_helper_cpu(const Tensor& self, Tensor& values, Tensor& indices, int64_t dim) { in cummax_helper_cpu()
796 std::tuple<Tensor&, Tensor&> cummax_out(const Tensor& self, int64_t dim, Tensor& values, Tensor& in… in cummax_out()
816 std::tuple<Tensor, Tensor> cummax(const Tensor& self, int64_t dim) { in cummax()
823 void cummin_helper_cpu(const Tensor& self, Tensor& values, Tensor& indices, int64_t dim) { in cummin_helper_cpu()
831 std::tuple<Tensor&, Tensor&> cummin_out(const Tensor& self, int64_t dim, Tensor& values, Tensor& in… in cummin_out()
851 std::tuple<Tensor, Tensor> cummin(const Tensor& self, int64_t dim) { in cummin()
858 …r cummaxmin_backward(const Tensor& grad, const Tensor& input, const Tensor& indices, int64_t dim) { in cummaxmin_backward()
872 …or& self, const std::optional<Tensor>& prepend, const std::optional<Tensor>& append, int64_t dim) { in prepend_append_on_dim()
884 …d diff_check_compatible_shape(const Tensor& self, const std::optional<Tensor>&other, int64_t dim) { in diff_check_compatible_shape()
908 static inline void diff_check(const Tensor& self, int64_t n, int64_t dim, const std::optional<Tenso… in diff_check()
922 static inline Tensor diff_helper(const Tensor& self, int64_t n, int64_t dim) { in diff_helper()
949 Tensor diff(const Tensor& self, int64_t n, int64_t dim, const std::optional<Tensor>& prepend, const… in diff()
959 static inline Tensor& diff_out_helper(const Tensor& self, int64_t n, int64_t dim, Tensor& result) { in diff_out_helper()
993 Tensor& diff_out(const Tensor& self, int64_t n, int64_t dim, const std::optional<Tensor>& prepend, … in diff_out()
1003 …dient(const Tensor& self, std::optional<int64_t> spacing_size, at::OptionalIntArrayRef dim, int64… in pre_check_gradient()
1033 …<Tensor> gradient_helper(const Tensor& self, TensorList coordinates, IntArrayRef dim, int64_t edge… in gradient_helper()
1074 … gradient_helper_float(const Tensor& self, ArrayRef<Scalar> spacing, IntArrayRef dim, int64_t edge… in gradient_helper_float()
1094 static std::vector<int64_t> gradient_dim_preprocess(const Tensor& self, std::optional<int64_t> dim)… in gradient_dim_preprocess()
1107 std::vector<Tensor> gradient(const Tensor& self, TensorList coordinates, IntArrayRef dim, int64_t e… in gradient()
1115 std::vector<Tensor> gradient(const Tensor& self, TensorList coordinates, std::optional<int64_t> dim… in gradient()
1124 std::vector<Tensor> gradient(const Tensor& self, c10::ArrayRef<Scalar> spacing, IntArrayRef dim, in… in gradient()
1132 …Tensor> gradient(const Tensor& self, ArrayRef<Scalar> spacing, std::optional<int64_t> dim, int64_t… in gradient()
1141 std::vector<Tensor> gradient(const Tensor& self, const Scalar& unit_size, IntArrayRef dim, int64_t … in gradient()
1152 …nt(const Tensor& self, const std::optional<Scalar>& unit_size, std::optional<int64_t> dim, int64_t… in gradient()
1166 std::vector<Tensor> gradient(const Tensor& self, IntArrayRef dim, int64_t edge_order) { in gradient()
1189 for (const auto dim : c10::irange(0, 2)) { in should_use_acc_buffer() local
1227 Tensor sum(const Tensor& self, DimnameList dim, bool keepdim, std::optional<ScalarType> dtype) { in sum()
1231 Tensor& sum_out(const Tensor& self, DimnameList dim, in sum_out()
1236 Tensor& nansum_out(const Tensor& self, at::OptionalIntArrayRef dim, in nansum_out()
1258 Tensor nansum(const Tensor& self, at::OptionalIntArrayRef dim, bool keepdim, std::optional<ScalarTy… in nansum()
1339 Tensor prod(const Tensor& self, Dimname dim, bool keepdim, std::optional<ScalarType> dtype) { in prod()
1343 Tensor& prod_out(const Tensor& self, Dimname dim, in prod_out()
1364 auto dim = opt_dim.value(); in TORCH_IMPL_FUNC() local
1408 Tensor mean(const Tensor& self, DimnameList dim, bool keepdim, std::optional<ScalarType> dtype) { in mean()
1412 Tensor& mean_out(const Tensor& self, DimnameList dim, in mean_out()
1429 at::OptionalIntArrayRef dim, in nanmean_out()
1444 at::OptionalIntArrayRef dim, in nanmean()
1524 IntArrayRef dim, in impl_func_norm()
1556 IntArrayRef dim, in sparse_norm()
1564 IntArrayRef dim, in sparse_dtype_norm()
1640 Tensor allany_dims_default(const Tensor &self, OptionalIntArrayRef dim, bool keepdim) { in allany_dims_default()
1678 Tensor all_dims_default(const Tensor &self, OptionalIntArrayRef dim, bool keepdim) { in all_dims_default()
1682 Tensor any_dims_default(const Tensor &self, OptionalIntArrayRef dim, bool keepdim) { in any_dims_default()
1687 const Tensor &self, OptionalIntArrayRef dim, bool keepdim, Tensor &result) { in all_dims_out_default()
1695 const Tensor &self, OptionalIntArrayRef dim, bool keepdim, Tensor &result) { in any_dims_out_default()
1721 std::optional<int64_t> dim, in argmax_argmin_impl()
1830 at::OptionalIntArrayRef dim, const std::optional<Scalar>& correction_opt, in std_var_out()
1902 at::OptionalIntArrayRef dim, const std::optional<Scalar>& correction_opt, in std_var_mean_out()
1973 const Tensor& self, at::OptionalIntArrayRef dim, bool unbiased, bool keepdim) { in var_mean()
1981 const Tensor& self, at::OptionalIntArrayRef dim, bool unbiased, bool keepdim) { in std_mean()
2000 Tensor& result1, Tensor& result2, const Tensor& self, IntArrayRef dim, in var_mean_out()
2012 const Tensor& self, at::OptionalIntArrayRef dim, in var_mean()
2021 const Tensor& self, at::OptionalIntArrayRef dim, in std_mean()
2035 Tensor var(const Tensor& self, at::OptionalIntArrayRef dim, bool unbiased, bool keepdim) { in var()
2042 Tensor& var_out(const Tensor& self, at::OptionalIntArrayRef dim, bool unbiased, bool keepdim, Tenso… in var_out()
2054 Tensor std(const Tensor& self, at::OptionalIntArrayRef dim, bool unbiased, bool keepdim) { in std()
2064 Tensor std(const Tensor& self, at::OptionalIntArrayRef dim, in std()
2071 const Tensor& self, at::OptionalIntArrayRef dim, in std_out()
2077 const Tensor& self, at::OptionalIntArrayRef dim, in var_out()
2083 const Tensor& self, at::OptionalIntArrayRef dim, in var()
2089 Tensor std(const Tensor& self, DimnameList dim, bool unbiased, bool keepdim) { in std()
2093 Tensor& std_out(const Tensor& self, DimnameList dim, bool unbiased, bool keepdim, Tensor& result) { in std_out()
2097 Tensor var(const Tensor& self, DimnameList dim, bool unbiased, bool keepdim) { in var()
2101 Tensor& var_out(const Tensor& self, DimnameList dim, bool unbiased, bool keepdim, Tensor& result) { in var_out()
2106 std::tuple<Tensor,Tensor> var_mean(const Tensor& self, DimnameList dim, bool unbiased, bool keepdim… in var_mean()
2110 std::tuple<Tensor,Tensor> std_mean(const Tensor& self, DimnameList dim, bool unbiased, bool keepdim… in std_mean()
2114 Tensor std(const Tensor& self, DimnameList dim, const std::optional<Scalar>& correction, bool keepd… in std()
2118 Tensor& std_out(const Tensor& self, DimnameList dim, const std::optional<Scalar>& correction, in std_out()
2123 Tensor var(const Tensor& self, DimnameList dim, const std::optional<Scalar>& correction, bool keepd… in var()
2127 Tensor& var_out(const Tensor& self, DimnameList dim, const std::optional<Scalar>& correction, in var_out()
2133 std::tuple<Tensor,Tensor> var_mean(const Tensor& self, DimnameList dim, in var_mean()
2138 std::tuple<Tensor,Tensor> std_mean(const Tensor& self, DimnameList dim, in std_mean()
2143 Tensor& norm_out(const Tensor& self, const std::optional<Scalar>& p, DimnameList dim, bool keepdim,… in norm_out()
2147 Tensor& norm_out(const Tensor& self, const std::optional<Scalar>& p, DimnameList dim, bool keepdim,… in norm_out()
2151 Tensor norm(const Tensor& self, const std::optional<Scalar>& p, DimnameList dim, bool keepdim, Scal… in norm()
2155 Tensor norm(const Tensor& self, const std::optional<Scalar>& p, DimnameList dim, bool keepdim) { in norm()
2159 Tensor any(const Tensor& self, Dimname dim, bool keepdim) { in any()
2162 Tensor& any_out(const Tensor &self, Dimname dim, bool keepdim, Tensor& result) { in any_out()
2165 Tensor all(const Tensor& self, Dimname dim, bool keepdim) { in all()
2168 Tensor& all_out(const Tensor &self, Dimname dim, bool keepdim, Tensor& result) { in all_out()
2179 Tensor logcumsumexp(const Tensor& self, Dimname dim) { in logcumsumexp()
2182 Tensor& logcumsumexp_out(const Tensor& self, Dimname dim, Tensor& result) { in logcumsumexp_out()
2185 Tensor cumsum(const Tensor& self, Dimname dim, std::optional<ScalarType> dtype) { in cumsum()
2188 Tensor& cumsum_(Tensor& self, Dimname dim, std::optional<ScalarType> dtype) { in cumsum_()
2191 Tensor& cumsum_out(const Tensor& self, Dimname dim, std::optional<ScalarType> dtype, Tensor& result… in cumsum_out()
2194 Tensor cumprod(const Tensor& self, Dimname dim, std::optional<ScalarType> dtype) { in cumprod()
2197 Tensor& cumprod_(Tensor& self, Dimname dim, std::optional<ScalarType> dtype) { in cumprod_()
2200 Tensor& cumprod_out(const Tensor& self, Dimname dim, std::optional<ScalarType> dtype, Tensor& resul… in cumprod_out()
2203 std::tuple<Tensor, Tensor> cummax(const Tensor& self, Dimname dim) { in cummax()
2206 std::tuple<Tensor&, Tensor&> cummax_out(const Tensor& self, Dimname dim, Tensor& values, Tensor& in… in cummax_out()
2209 std::tuple<Tensor, Tensor> cummin(const Tensor& self, Dimname dim) { in cummin()
2212 std::tuple<Tensor&, Tensor&> cummin_out(const Tensor& self, Dimname dim, Tensor& values, Tensor& in… in cummin_out()
2298 Tensor value_selecting_reduction_backward_symint(const Tensor& grad, int64_t dim, const Tensor& ind… in value_selecting_reduction_backward_symint()
2324 Tensor sum_sparse_coo(const Tensor& self, at::OptionalIntArrayRef dim, bool keepdim, std::optional<… in sum_sparse_coo()
2341 for (int dim = 0; dim < self.dim(); dim++) { in sum_sparse_coo() local
2352 at::OptionalIntArrayRef dim, in sum_sparse_compressed()