Spaces:
Running
Running
File size: 1,897 Bytes
28c256d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
// Copyright (c) OpenMMLab. All rights reserved
#include "pytorch_cpp_helper.hpp"
#include "pytorch_device_registry.hpp"
void bezier_align_forward_impl(Tensor input, Tensor rois, Tensor output,
int aligned_height, int aligned_width,
float spatial_scale, int sampling_ratio,
bool aligned) {
DISPATCH_DEVICE_IMPL(bezier_align_forward_impl, input, rois, output,
aligned_height, aligned_width, spatial_scale,
sampling_ratio, aligned);
}
void bezier_align_backward_impl(Tensor grad_output, Tensor rois,
Tensor grad_input, int aligned_height,
int aligned_width, float spatial_scale,
int sampling_ratio, bool aligned) {
DISPATCH_DEVICE_IMPL(bezier_align_backward_impl, grad_output, rois,
grad_input, aligned_height, aligned_width, spatial_scale,
sampling_ratio, aligned);
}
void bezier_align_forward(Tensor input, Tensor rois, Tensor output,
int aligned_height, int aligned_width,
float spatial_scale, int sampling_ratio,
bool aligned) {
bezier_align_forward_impl(input, rois, output, aligned_height, aligned_width,
spatial_scale, sampling_ratio, aligned);
}
void bezier_align_backward(Tensor grad_output, Tensor rois, Tensor grad_input,
int aligned_height, int aligned_width,
float spatial_scale, int sampling_ratio,
bool aligned) {
bezier_align_backward_impl(grad_output, rois, grad_input, aligned_height,
aligned_width, spatial_scale, sampling_ratio,
aligned);
}
|