mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
3786856660
11 lines
411 B
C++
11 lines
411 B
C++
// Copyright (c) OpenMMLab. All rights reserved.
|
|||
#include "torch/script.h"
|
|||
|
|||
TORCH_LIBRARY(mmdeploy, m) {
|
|||
m.def(
|
|||
"modulated_deform_conv(Tensor input, Tensor weight, Tensor bias, Tensor offset, Tensor "
|
|||
"mask, "
|
|||
"int kernel_h, int kernel_w, int stride_h, int stride_w, int pad_h, int pad_w, int "
|
|||
"dilation_h,int dilation_w, int groups, int deform_groups, bool with_bias) -> Tensor");
|
|||
}
|