Searched defs:x2_norm (Results 1 – 1 of 1) sorted by relevance
72 Tensor x2_norm = x2.pow(2).sum(-1, true); in _euclidean_dist() local321 auto x2_norm = at::linalg_vector_norm(*x2, 2, /*dim=*/dim, /*keepdim=*/true).clone(); in cosine_similarity() local