Home
last modified time | relevance | path

Searched refs:src_expanded (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DFunctionsManual.cpp4629 auto src_expanded = src; in unsqueeze_dim1() local
4630 while (src_expanded.sizes().size() < target.sizes().size() - 1) { in unsqueeze_dim1()
4631 src_expanded = src_expanded.unsqueeze(1); in unsqueeze_dim1()
4633 if (src_expanded.sizes().size() == target.sizes().size() - 1) { in unsqueeze_dim1()
4634 src_expanded = src_expanded.unsqueeze(0); in unsqueeze_dim1()
4636 return src_expanded; in unsqueeze_dim1()
4643 auto src_expanded = src; in expand_as_dim1() local
4644 while (src_expanded.sizes().size() < target.sizes().size() - 1) { in expand_as_dim1()
4645 src_expanded = src_expanded.unsqueeze(1); in expand_as_dim1()
4647 return src_expanded.expand_as(target); in expand_as_dim1()