Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DDistance.cpp126 …Tensor tensor1_expanded = x1.expand_symint(tensor1_expand_size).contiguous().view_symint(tensor1_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 DLinearAlgebra.cpp2129 const auto tensor1_expanded = tensor1.expand(tensor1_expand_size) in _matmul_impl() local
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 Ddecompositions.py4405 tensor1_expanded = tensor1.expand(tensor1_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)