Searched refs:tensor1_expand_size (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Distance.cpp | 117 std::vector<SymInt> tensor1_expand_size(expand_batch_portion); in cdist_impl() local
118 tensor1_expand_size.insert(tensor1_expand_size.end(), {r1, c1}); in cdist_impl()
126 …Tensor tensor1_expanded = x1.expand_symint(tensor1_expand_size).contiguous().view_symint(tensor1_v… in cdist_impl()
202 std::vector<int64_t> tensor1_expand_size(expand_batch_portion); in _cdist_backward() local
203 tensor1_expand_size.insert(tensor1_expand_size.end(), {r1, c1}); in _cdist_backward()
216 if (tensor1_expand_size != x1.sizes()) { in _cdist_backward()
217 x1 = x1.expand(tensor1_expand_size); in _cdist_backward()
|
| H A D | LinearAlgebra.cpp | 2126 const auto tensor1_expand_size = [&output_shape, n, m1]{ DimVector ret(output_shape); in _matmul_impl() local
2129 const auto tensor1_expanded = tensor1.expand(tensor1_expand_size) in _matmul_impl()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/ |
| H A D | ReduceOps.mm | 1095 std::vector<int64_t> tensor1_expand_size(expand_batch_portion);
1096 tensor1_expand_size.insert(tensor1_expand_size.end(), {r1, c1});
1112 toShape:getMPSShape(tensor1_expand_size)
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 3334 tensor1_expand_size = expand_batch_portion.copy()
3335 tensor1_expand_size.extend([r1, c1])
3339 if tensor1_expand_size != list(x1.shape):
3340 x1 = x1.expand(tensor1_expand_size)
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 4400 tensor1_expand_size = expand_batch_portion + [n, m1]
4405 tensor1_expanded = tensor1.expand(tensor1_expand_size).reshape(
|