InterpolateKernel.h File Reference

(252c867 (Sun May 5 05:39:34 2024 +0000))

Go to the source code of this file.

Functions
void	three_nn_launcher (int b, int n, int m, const float *unknown, const float *known, float *dist2, int *idx)
void	three_interpolate_launcher (int b, int c, int m, int n, const float *points, const int *idx, const float *weight, float *out)
void	three_interpolate_grad_launcher (int b, int c, int n, int m, const float *grad_out, const int *idx, const float *weight, float *grad_points)

Function Documentation

three_interpolate_grad_launcher()

void three_interpolate_grad_launcher	(	int	b,
		int	c,
		int	n,
		int	m,
		const float *	grad_out,
		const int *	idx,
		const float *	weight,
		float *	grad_points
	)

three_interpolate_launcher()

void three_interpolate_launcher	(	int	b,
		int	c,
		int	m,
		int	n,
		const float *	points,
		const int *	idx,
		const float *	weight,
		float *	out
	)

three_nn_launcher()

void three_nn_launcher	(	int	b,
		int	n,
		int	m,
		const float *	unknown,
		const float *	known,
		float *	dist2,
		int *	idx
	)