NmsOpKernel.cpp File Reference

(b9e049c (Sun Dec 31 11:36:26 2023 -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);