11 lines
386 B
C
11 lines
386 B
C
|
#pragma once
|
||
|
|
|
||
|
|
#include <torch/extension.h>
|
||
|
|
#include <vector>
|
||
|
|
|
||
|
|
std::vector<at::Tensor> three_nn(at::Tensor unknowns, at::Tensor knows);
|
||
|
|
at::Tensor three_interpolate(at::Tensor points, at::Tensor idx,
|
||
|
|
at::Tensor weight);
|
||
|
|
at::Tensor three_interpolate_grad(at::Tensor grad_out, at::Tensor idx,
|
||
|
|
at::Tensor weight, const int m);
|