ReduceSubarraysSumKernelCPU.cpp File Reference

#include "tensorflow/core/framework/op.h"
#include "tensorflow/core/framework/op_kernel.h"
#include "tensorflow/core/lib/core/errors.h"
#include "Open3D/ML/Misc/Detail/ReduceSubarraysSumCPU.h"

Data Structures
class	ReduceSubarraysSumOp< T >

Macros
#define	REG_KB(type)

Macro Definition Documentation

REG_KB

#define REG_KB

(

type

)

Value:

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