| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | TensorIndexing.h | 210 const std::optional<SymIntArrayRef>& self_sizes) { in applySlice()
240 const std::optional<SymIntArrayRef>& self_sizes) { in applySelect()
520 const std::optional<SymIntArrayRef>& self_sizes) { in applySlicing()
612 std::optional<SymIntArrayRef> self_sizes = self.is_nested() variable
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_expand_copy.cpp | 31 exec_aten::ArrayRef<SizesType> self_sizes, in map_expand_to_repeats()
70 const auto& self_sizes = self.sizes(); in expand_copy_out() local
|
| H A D | op_constant_pad_nd.cpp | 34 IntArrayRef self_sizes, in apply_padding_to_dim()
119 int64_t self_sizes[kTensorDimensionLimit]; in constant_pad_nd_out_impl() local
|
| H A D | op_repeat.cpp | 21 const exec_aten::ArrayRef<exec_aten::SizesType>& self_sizes, in calculate_output_size()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorMatmul.cpp | 82 std::vector<IntArrayRef> self_sizes = NestedTensor_get_sizes(self_ptr); in matmul_with_bmm_nested() local
259 const auto& self_sizes = self_ptr->get_nested_sizes(); in matmul_nested() local
|
| H A D | NestedTensorBackward.cpp | 128 auto self_sizes = nt_self->get_nested_sizes(); in _nested_sum_backward_cpu() local
170 const auto self_sizes = nt_self->get_nested_sizes(); in _nested_select_backward_symint() local
|
| H A D | NestedTensorTransformerFunctions.cpp | 131 const auto& self_sizes = nt_self.get_nested_sizes(); in NestedTensor_add_NestedTensor_in_place() local
|
| H A D | NestedTensorBinaryOps.cpp | 131 const auto self_sizes = self_ptr->get_nested_sizes(); in NestedTensor_elementwise_Tensor() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorAdvancedIndexing.cpp | 346 auto self_sizes = self.sizes().vec(); in index_func_meta_impl() local
1256 auto self_sizes = self_contig.sizes(); in index_select_out_cpu_dim1_() local
1475 Tensor index_select_backward_symint(const Tensor& grad, c10::SymIntArrayRef self_sizes, int64_t dim… in index_select_backward_symint()
1527 auto self_sizes = self_nonzero_dim.sizes().vec(); in index_fill_() local
2169 SymDimVector self_sizes{self.sym_sizes()}; in _take_along_dim_helper() local
2368 const auto self_sizes = self.sizes(); in nonzero_out_cpu() local
|
| H A D | AutogradComposite.cpp | 56 auto self_sizes = self.sym_sizes(); in _new_zeros_with_same_feature_meta() local
|
| H A D | LegacyBatching.cpp | 108 auto self_sizes = self.sizes(); in _remove_batch_dim() local
|
| H A D | ReduceOpsUtils.h | 45 IntArrayRef self_sizes = self.sizes(); in _dimreduce_setup() local
|
| H A D | Normalization.cpp | 943 auto self_sizes = self.sizes(); in TORCH_IMPL_FUNC() local
|
| H A D | LinearAlgebra.cpp | 315 const auto self_sizes = self.sizes(); in common_checks_baddbmm_bmm() local
|
| H A D | TensorShape.cpp | 1297 const auto self_sizes = self_contig->sizes(); in narrow_copy_dense_cpu_out() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | TensorModeKernel.cpp | 22 auto self_sizes = ensure_nonempty_vec(self.sizes().vec()); in mode_kernel_impl() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseBlas.cpp | 256 IntArrayRef self_sizes = self.sizes(); in sparse_sampled_addmm_check_inputs() local
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | Linear.cpp | 32 std::vector<int64_t> self_sizes = graph.sizes_of(self); in check_addmm_args() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | PyTorchOperatorHacks.cpp | 30 Tensor index_select_backward_hack(const Tensor& grad, IntArrayRef self_sizes, int64_t dim, const Te… in index_select_backward_hack()
|
| H A D | BatchRulesViews.cpp | 453 auto self_sizes = self_.sym_sizes(); in expand_batch_rule() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/functorch/ |
| H A D | init.cpp | 153 auto self_sizes = self.sizes(); in _remove_batch_dim() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | TensorCompareKernel.cpp | 39 auto self_sizes = ensure_nonempty_vec(self.sizes().vec()); in compare_base_kernel_core() local
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | copy_ops_util.cpp | 184 exec_aten::ArrayRef<exec_aten::SizesType> self_sizes, in get_expand_copy_out_target_size()
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | shape_inference.cpp | 490 auto self_sizes = self.sizes(); in compute_shape_index_select() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 1644 auto self_sizes = self_p.sizes(); in renorm_jvp() local
|