HoLa-BRep

Runtime error

HoLa-BRep / pointnet2_ops_lib /pointnet2_ops /_ext-src /include /interpolate.h

fresh start

e289681 about 1 year ago

396 Bytes

#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);

	#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);