Searched refs:tensor2_expanded (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Distance.cpp | 127 …Tensor tensor2_expanded = x2.expand_symint(tensor2_expand_size).contiguous().view_symint(tensor2_v… in cdist_impl() local
141 at::_euclidean_dist(tensor1_expanded, tensor2_expanded); in cdist_impl()
145 cdist_stub(device1, result, tensor1_expanded, tensor2_expanded, p); in cdist_impl()
|
| H A D | LinearAlgebra.cpp | 2143 auto tensor2_expanded = tensor2.expand(tensor2_expand_size); in _matmul_impl() local
2145 tensor2_expanded = tensor2_expanded.reshape({expand_batch_product, m2}).unsqueeze(2); in _matmul_impl()
2147 tensor2_expanded = tensor2_expanded.reshape({expand_batch_product, m2, p}); in _matmul_impl()
2159 return at::_unsafe_view(tensor1_expanded.bmm(tensor2_expanded).squeeze(-1), output_shape); in _matmul_impl()
2161 return at::_unsafe_view(tensor1_expanded.bmm(tensor2_expanded), output_shape); in _matmul_impl()
2166 at::bmm_out(reshaped_out, tensor1_expanded, tensor2_expanded); in _matmul_impl()
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 4412 tensor2_expanded = (
4419 tensor2_expanded = tensor2.expand(tensor2_expand_size).reshape(
4431 return tensor1_expanded.bmm(tensor2_expanded).squeeze(-1).view(output_shape)
4433 return tensor1_expanded.bmm(tensor2_expanded).view(output_shape)
|