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