Searched defs:tensor2_expanded (Results 1 – 2 of 2) sorted by relevance
127 …Tensor tensor2_expanded = x2.expand_symint(tensor2_expand_size).contiguous().view_symint(tensor2_v… in cdist_impl() local
2143 auto tensor2_expanded = tensor2.expand(tensor2_expand_size); in _matmul_impl() local