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