File size: 1,100 Bytes
e05eed1 98a67a0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
// SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
// SPDX-License-Identifier: Apache-2.0
#include <torch/torch.h>
torch::Tensor quad_rectify_gpu_calc_quad_width(torch::Tensor quads,
int64_t outputHeight,
int64_t roundFactor,
float maxWidth);
torch::Tensor quad_rectify_gpu_forward(torch::Tensor quads,
int64_t imageHeight,
int64_t imageWidth,
int64_t outputHeight,
int64_t outputWidth,
bool isotropic);
torch::Tensor quad_rectify_gpu_backward(torch::Tensor quads,
torch::Tensor gradOutput,
int64_t imageHeight,
int64_t imageWidth,
bool isotropic);
|