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