SparseConvOpKernel.cpp File Reference

(b9e049c (Sun Dec 31 11:36:26 2023 -0800))

#include "SparseConvOpKernel.h"
#include "open3d/ml/impl/sparse_conv/SparseConv.h"

Data Structures
class	SparseConvOpKernelCPU< TFeat, TOut, TIndex, TKernelIndex >

Macros
#define	REG_KB(feattype, outtype, indextype, kernelindextype)

Macro Definition Documentation

REG_KB

#define REG_KB	(		feattype,
			outtype,
			indextype,
			kernelindextype
	)

Value:

    REGISTER_KERNEL_BUILDER(                                          \
            Name("Open3DSparseConv")                                  \
                    .Device(DEVICE_CPU)                               \
                    .TypeConstraint<feattype>("TFeat")                \
                    .TypeConstraint<outtype>("output_type")           \
                    .TypeConstraint<indextype>("TIndex")              \
                    .TypeConstraint<kernelindextype>("TKernelIndex"), \
            SparseConvOpKernelCPU<feattype, outtype, indextype,       \
                                  kernelindextype>);