NmsOpKernel.cpp File Reference

(313315d (Wed Dec 23 23:39:47 2020 -0800))

#include "open3d/ml/tensorflow/misc/NmsOpKernel.h"
#include "open3d/ml/contrib/Nms.h"

Data Structures
class	NmsOpKernelCPU

Macros
#define	REG_KB(type)

Macro Definition Documentation

REG_KB

#define REG_KB

(

type

)

Value:

REGISTER_KERNEL_BUILDER(                                                \
            Name("Open3DNms").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
            NmsOpKernelCPU);