File size: 437 Bytes
e05eed1 98a67a0 |
1 2 3 4 5 6 7 8 9 10 11 |
// SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
// SPDX-License-Identifier: Apache-2.0
#pragma once
#include <torch/torch.h>
torch::Tensor ragged_quad_all_2_all_distance_v2(torch::Tensor embedQuads, torch::Tensor quadsPerExample,
float xFactor, float yFactor,
bool allowSelfDistance);
|