Searched refs:x1_pad (Results 1 – 2 of 2) sorted by relevance
71 Tensor x1_pad = at::ones_like(x1_norm, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in _euclidean_dist() local74 Tensor x1_ = at::cat({x1.mul(-2), std::move(x1_norm), std::move(x1_pad)}, -1); in _euclidean_dist()
718 x1_pad = torch.ones_like(x1_norm, memory_format=torch.contiguous_format)721 x1_ = torch.cat([x1.mul(-2), x1_norm, x1_pad], -1)