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