Home
last modified time | relevance | path

Searched defs:num_new_dimensions (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dshape_inference.cpp432 size_t num_new_dimensions = size.size() - self.dim(); in compute_shape_expand() local
449 size_t num_new_dimensions = _sizes.size() - self.dim(); in compute_shape_expand() local
1122 size_t num_new_dimensions = repeats.size() - self.dim(); in compute_shape_repeat() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dops.cpp64 int64_t num_new_dimensions = repeats.size() - self.dim(); in repeat_out() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp1513 int64_t num_new_dimensions = repeats.size() - self.dim(); in repeat() local