#include <InterpolateOpKernel.h>
|
| ThreeInterpolateOpKernel (tensorflow::OpKernelConstruction *construction) |
|
void | Compute (tensorflow::OpKernelContext *context) override |
|
virtual void | Kernel (tensorflow::OpKernelContext *context, int b, int c, int m, int n, const float *points, const int *idx, const float *weight, float *out)=0 |
|
◆ ThreeInterpolateOpKernel()
ThreeInterpolateOpKernel::ThreeInterpolateOpKernel |
( |
tensorflow::OpKernelConstruction * |
construction | ) |
|
|
inlineexplicit |
◆ Compute()
void ThreeInterpolateOpKernel::Compute |
( |
tensorflow::OpKernelContext * |
context | ) |
|
|
inlineoverride |
◆ Kernel()
virtual void ThreeInterpolateOpKernel::Kernel |
( |
tensorflow::OpKernelContext * |
context, |
|
|
int |
b, |
|
|
int |
c, |
|
|
int |
m, |
|
|
int |
n, |
|
|
const float * |
points, |
|
|
const int * |
idx, |
|
|
const float * |
weight, |
|
|
float * |
out |
|
) |
| |
|
pure virtual |
The documentation for this class was generated from the following file: