Searched defs:dim_opt (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | Repeat.cpp | 104 std::optional<int64_t> dim_opt, in repeat_interleave_symint()
|
| D | ReduceOpsUtils.h | 187 at::OptionalIntArrayRef dim_opt, in make_reduction()
227 at::OptionalIntArrayRef dim_opt, bool keepdim, ScalarType dtype1, in make_reduction()
|
| D | SpectralOps.cpp | 753 static DimVector default_alldims(const Tensor& self, at::OptionalIntArrayRef dim_opt) { in default_alldims()
768 Tensor fft_fftshift(const Tensor& x, at::OptionalIntArrayRef dim_opt) { in fft_fftshift()
780 Tensor fft_ifftshift(const Tensor& x, at::OptionalIntArrayRef dim_opt) { in fft_ifftshift()
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | concat_opt.cpp | 542 const auto dim_opt = toIValue(node->input(1)); in handleConcat() local
|
| /external/pytorch/torch/csrc/jit/runtime/static/ |
| D | passes.cpp | 1265 auto dim_opt = toIValue(squeeze_node->input(1)); in UseSplitAndSqueeze() local
|
| /external/pytorch/torch/csrc/autograd/ |
| D | FunctionsManual.cpp | 1832 at::OptionalIntArrayRef dim_opt, in var_backward()
1876 at::OptionalIntArrayRef dim_opt, in var_mean_backward()
1900 at::OptionalIntArrayRef dim_opt, in std_mean_backward()
|