#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: