Searched refs:tensor1_expanded (Results 1 – 3 of 3) sorted by relevance
126 …Tensor tensor1_expanded = x1.expand_symint(tensor1_expand_size).contiguous().view_symint(tensor1_v… in cdist_impl() local141 at::_euclidean_dist(tensor1_expanded, tensor2_expanded); in cdist_impl()145 cdist_stub(device1, result, tensor1_expanded, tensor2_expanded, p); in cdist_impl()
2129 const auto tensor1_expanded = tensor1.expand(tensor1_expand_size) in _matmul_impl() local2159 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()
4405 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)