InterpolateKernelCPU.cpp File Reference

(d32b4fc (Mon Sep 14 15:46:25 2026 -0700))

#include <tbb/blocked_range.h>
#include <tbb/parallel_for.h>
#include <algorithm>
#include <cmath>
#include <limits>
#include "open3d/ml/pytorch/pointnet/InterpolateKernel.h"

Functions
void	three_nn_launcher_cpu (int b, int n, int m, const float *unknown, const float *known, float *dist2, int *idx)
void	three_interpolate_launcher_cpu (int b, int c, int m, int n, const float *points, const int *idx, const float *weight, float *out)
void	three_interpolate_grad_launcher_cpu (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_cpu()

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

three_interpolate_launcher_cpu()

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

three_nn_launcher_cpu()

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