Searched refs:src_expanded (Results 1 – 1 of 1) sorted by relevance
4629 auto src_expanded = src; in unsqueeze_dim1() local4630 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() local4644 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()