Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/lazy/core/
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
/external/pytorch/torch/csrc/jit/runtime/static/
Dops.cpp64 int64_t num_new_dimensions = repeats.size() - self.dim(); in repeat_out() local
/external/pytorch/aten/src/ATen/native/
DTensorShape.cpp1513 int64_t num_new_dimensions = repeats.size() - self.dim(); in repeat() local