#include <RoiPoolOpKernel.h>
|
| RoiPoolOpKernel (tensorflow::OpKernelConstruction *construction) |
|
void | Compute (tensorflow::OpKernelContext *context) override |
|
virtual void | Kernel (tensorflow::OpKernelContext *context, int batch_size, int pts_num, int boxes_num, int feature_in_len, int sampled_pts_num, const float *xyz, const float *boxes3d, const float *pts_feature, float *pooled_features, int *pooled_empty_flag)=0 |
|
◆ RoiPoolOpKernel()
RoiPoolOpKernel::RoiPoolOpKernel |
( |
tensorflow::OpKernelConstruction * |
construction | ) |
|
|
inlineexplicit |
◆ Compute()
void RoiPoolOpKernel::Compute |
( |
tensorflow::OpKernelContext * |
context | ) |
|
|
inlineoverride |
◆ Kernel()
virtual void RoiPoolOpKernel::Kernel |
( |
tensorflow::OpKernelContext * |
context, |
|
|
int |
batch_size, |
|
|
int |
pts_num, |
|
|
int |
boxes_num, |
|
|
int |
feature_in_len, |
|
|
int |
sampled_pts_num, |
|
|
const float * |
xyz, |
|
|
const float * |
boxes3d, |
|
|
const float * |
pts_feature, |
|
|
float * |
pooled_features, |
|
|
int * |
pooled_empty_flag |
|
) |
| |
|
pure virtual |
◆ sampled_pts_num
int RoiPoolOpKernel::sampled_pts_num |
|
protected |
The documentation for this class was generated from the following file: